YinuoGuo27 commited on
Commit
a4c475a
·
verified ·
1 Parent(s): 89688fa

Update difpoint/inference.py

Browse files
Files changed (1) hide show
  1. difpoint/inference.py +1 -1
difpoint/inference.py CHANGED
@@ -178,7 +178,7 @@ class Inferencer(object):
178
  self.wav2lip_model.eval()
179
 
180
  # specify configs for inference
181
- self.inf_cfg = OmegaConf.load("difpoint/configs/onnx_infer.yaml")
182
  self.inf_cfg.infer_params.flag_pasteback = False
183
 
184
  self.live_portrait_pipeline = FasterLivePortraitPipeline(cfg=self.inf_cfg, is_animal=False)
 
178
  self.wav2lip_model.eval()
179
 
180
  # specify configs for inference
181
+ self.inf_cfg = OmegaConf.load("difpoint/configs/onnx_mp_infer.yaml")
182
  self.inf_cfg.infer_params.flag_pasteback = False
183
 
184
  self.live_portrait_pipeline = FasterLivePortraitPipeline(cfg=self.inf_cfg, is_animal=False)