Spaces:
Sleeping
Sleeping
Update src/facerender/animate.py
Browse files
src/facerender/animate.py
CHANGED
@@ -143,7 +143,7 @@ class AnimateFromCoeff():
|
|
143 |
|
144 |
return checkpoint['epoch']
|
145 |
|
146 |
-
|
147 |
|
148 |
def load_cpk_mapping(self, checkpoint_path, mapping=None, discriminator=None,
|
149 |
optimizer_mapping=None, optimizer_discriminator=None, device='cpu'):
|
|
|
143 |
|
144 |
return checkpoint['epoch']
|
145 |
|
146 |
+
|
147 |
|
148 |
def load_cpk_mapping(self, checkpoint_path, mapping=None, discriminator=None,
|
149 |
optimizer_mapping=None, optimizer_discriminator=None, device='cpu'):
|