Update difpoint/inference.py

difpoint/inference.py

@@ -183,7 +183,7 @@ class Inferencer(object):
         self.inf_cfg = OmegaConf.load("difpoint/configs/onnx_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,6 +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.croper = Croper(self.lm_croper_checkpoint)
         self.live_portrait_pipeline.init()
         if audio_type == 'upload':