YinuoGuo27 commited on
Commit
bcf7449
·
verified ·
1 Parent(s): 7aaeb95

Update difpoint/inference.py

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