Spaces:

moflo
/

nftGAN

Runtime error

d. nye commited on Jan 8, 2022

Commit

021195d

1 Parent(s): c68dd81

Error with checkpoints

Files changed (1) hide show

app.py CHANGED Viewed

@@ -591,7 +591,7 @@ class InferenceWrapper:
         self.model = model
         self.style_gan = StyleGAN(start_res=START_RES, target_res=TARGET_RES)
         self.style_gan.grow_model(64)
-        self.style_gan.load_weights(os.path.join("pretrained/checkpoints/stylegan_64x64.ckpt"))
         self.seed = -1
     def __call__(self, seed, feature):

         self.model = model
         self.style_gan = StyleGAN(start_res=START_RES, target_res=TARGET_RES)
         self.style_gan.grow_model(64)
+        self.style_gan.load_weights(os.path.join("pretrained/stylegan_128x128.ckpt"))
         self.seed = -1
     def __call__(self, seed, feature):