Spaces:
Paused
Paused
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",
|
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)
|