YinuoGuo27 commited on
Commit
50df1fb
·
verified ·
1 Parent(s): 11c707b

Update difpoint/inference.py

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