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 |
|