Spaces:

p1atdev
/

Anime-to-Sketch

Running

p1atdev commited on May 5, 2023

Commit

413b5ca

•

1 Parent(s): bfaed2e

fix: use half model

Files changed (1) hide show

anime2sketch/model.py CHANGED Viewed

@@ -179,7 +179,7 @@ class Anime2Sketch:
         for key in list(ckpt.keys()):
             if "module." in key:
-                ckpt[key.replace("module.", "")] = ckpt[key]
                 del ckpt[key]
         net.load_state_dict(ckpt)

         for key in list(ckpt.keys()):
             if "module." in key:
+                ckpt[key.replace("module.", "")] = ckpt[key].half()
                 del ckpt[key]
         net.load_state_dict(ckpt)