Spanicin commited on
Commit
0eee9b6
1 Parent(s): c65548e

Update videoretalking/inference_function.py

Browse files
videoretalking/inference_function.py CHANGED
@@ -25,7 +25,7 @@ from .utils.inference_utils import Laplacian_Pyramid_Blending_with_mask, face_de
25
  import warnings
26
  warnings.filterwarnings("ignore")
27
 
28
- def video_lipsync_correctness(face, audio_path, outfile=None, tmp_dir="temp", crop=[0, -1, 0, -1], re_preprocess=False, exp_img="neutral", face3d_net_path = face3d_net_path, one_shot=False, up_face="original", LNet_batch_size=16, without_rl1=False, static=False):
29
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
30
  print('[Info] Using {} for inference.'.format(device))
31
  os.makedirs(os.path.join('temp', tmp_dir), exist_ok=True)
 
25
  import warnings
26
  warnings.filterwarnings("ignore")
27
 
28
+ def video_lipsync_correctness(face, audio_path, face3d_net_path, outfile=None, tmp_dir="temp", crop=[0, -1, 0, -1], re_preprocess=False, exp_img="neutral", one_shot=False, up_face="original", LNet_batch_size=16, without_rl1=False, static=False):
29
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
30
  print('[Info] Using {} for inference.'.format(device))
31
  os.makedirs(os.path.join('temp', tmp_dir), exist_ok=True)