Update difpoint/inference.py

difpoint/inference.py

@@ -8,7 +8,7 @@ import glob
 
 import os
 os.environ['HYDRA_FULL_ERROR']='1'
-os.environ['CUDA_VISIBLE_DEVICES'] = '2'
+
 
 import os
 import time
@@ -183,7 +183,7 @@ class Inferencer(object):
         self.inf_cfg = OmegaConf.load("difpoint/configs/onnx_mp_infer.yaml")
         self.inf_cfg.infer_params.flag_pasteback = False
 
-        #self.live_portrait_pipeline = FasterLivePortraitPipeline(cfg=self.inf_cfg, is_animal=False)
+        self.live_portrait_pipeline = FasterLivePortraitPipeline(cfg=self.inf_cfg, is_animal=False)
         #ret = self.live_portrait_pipeline.prepare_source(source_image)
 
         print('#'*25+f'End initialization, cost time {time.time()-st}'+'#'*25)
@@ -274,7 +274,7 @@ class Inferencer(object):
     # 2024.06.26
     @torch.no_grad()
     def generate_with_audio_img(self, upload_audio_path, tts_audio_path, audio_type, image_path, smoothed_pitch, smoothed_yaw, smoothed_roll, smoothed_t,  save_path='results'):
-        self.live_portrait_pipeline = FasterLivePortraitPipeline(cfg=self.inf_cfg, is_animal=False)
+        #self.live_portrait_pipeline = FasterLivePortraitPipeline(cfg=self.inf_cfg, is_animal=False)
         if audio_type == 'upload':
             audio_path = upload_audio_path
         elif audio_type == 'tts':