d. nye
commited on
Commit
·
1a3d083
1
Parent(s):
28f8e40
Update model size to 128
Browse files
app.py
CHANGED
@@ -590,7 +590,7 @@ class InferenceWrapper:
|
|
590 |
def __init__(self, model):
|
591 |
self.model = model
|
592 |
self.style_gan = StyleGAN(start_res=START_RES, target_res=TARGET_RES)
|
593 |
-
self.style_gan.grow_model(
|
594 |
self.style_gan.load_weights(os.path.join("pretrained/stylegan_128x128.ckpt"))
|
595 |
self.seed = -1
|
596 |
|
|
|
590 |
def __init__(self, model):
|
591 |
self.model = model
|
592 |
self.style_gan = StyleGAN(start_res=START_RES, target_res=TARGET_RES)
|
593 |
+
self.style_gan.grow_model(128)
|
594 |
self.style_gan.load_weights(os.path.join("pretrained/stylegan_128x128.ckpt"))
|
595 |
self.seed = -1
|
596 |
|