Spaces:
Sleeping
Sleeping
Update difpoint/inference.py
Browse files- difpoint/inference.py +2 -2
difpoint/inference.py
CHANGED
@@ -166,12 +166,12 @@ class Inferencer(object):
|
|
166 |
self.point_diffusion.eval()
|
167 |
self.point_diffusion.to(self.device)
|
168 |
|
169 |
-
lm_croper_checkpoint =
|
170 |
self.croper = Croper(lm_croper_checkpoint)
|
171 |
|
172 |
self.norm_info = dict(np.load(r'difpoint/datasets/norm_info_d6.5_c8.5_vox1_train.npz'))
|
173 |
|
174 |
-
wav2lip_checkpoint = '
|
175 |
self.wav2lip_model = AudioEncoder(wav2lip_checkpoint, 'cuda')
|
176 |
self.wav2lip_model.cuda()
|
177 |
self.wav2lip_model.eval()
|
|
|
166 |
self.point_diffusion.eval()
|
167 |
self.point_diffusion.to(self.device)
|
168 |
|
169 |
+
lm_croper_checkpoint = './downloaded_repo/ckpts/shape_predictor_68_face_landmarks.dat'
|
170 |
self.croper = Croper(lm_croper_checkpoint)
|
171 |
|
172 |
self.norm_info = dict(np.load(r'difpoint/datasets/norm_info_d6.5_c8.5_vox1_train.npz'))
|
173 |
|
174 |
+
wav2lip_checkpoint = './downloaded_repo/ckpts/wav2lip.pth'
|
175 |
self.wav2lip_model = AudioEncoder(wav2lip_checkpoint, 'cuda')
|
176 |
self.wav2lip_model.cuda()
|
177 |
self.wav2lip_model.eval()
|