Spaces:
Sleeping
Sleeping
Update difpoint/inference.py
Browse files- difpoint/inference.py +1 -1
difpoint/inference.py
CHANGED
@@ -168,7 +168,7 @@ class Inferencer(object):
|
|
168 |
self.point_diffusion.eval()
|
169 |
self.point_diffusion.to(self.device)
|
170 |
|
171 |
-
|
172 |
self.croper = Croper(lm_croper_checkpoint)
|
173 |
|
174 |
self.norm_info = dict(np.load(r'difpoint/datasets/norm_info_d6.5_c8.5_vox1_train.npz'))
|
|
|
168 |
self.point_diffusion.eval()
|
169 |
self.point_diffusion.to(self.device)
|
170 |
|
171 |
+
lm_croper_checkpoint = './downloaded_repo/ckpts/shape_predictor_68_face_landmarks.dat'
|
172 |
self.croper = Croper(lm_croper_checkpoint)
|
173 |
|
174 |
self.norm_info = dict(np.load(r'difpoint/datasets/norm_info_d6.5_c8.5_vox1_train.npz'))
|