Spaces:

aijack
/

smile

Runtime error

aijack commited on Feb 26, 2023

Commit

804ca40

1 Parent(s): 4909045

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -80,7 +80,7 @@ class ImageEditor(object):
             checkpoint = torch.load(model_paths[model], map_location=self.device)
-            g_ema.load_state_dict(checkpoint['g_ema'])
             self.generators[model] = g_ema

             checkpoint = torch.load(model_paths[model], map_location=self.device)
+            g_ema.load_state_dict(checkpoint['g_ema'], strict=False)
             self.generators[model] = g_ema