YinuoGuo27 commited on
Commit
a68c679
·
verified ·
1 Parent(s): 47e2a88

Update difpoint/inference.py

Browse files
Files changed (1) hide show
  1. 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 = os.path.join('difpoint/dataset_process/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 = 'difpoint/dataset_process/ckpts/wav2lip.pth'
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()