Spaces:
Runtime error
Runtime error
moflo
commited on
Commit
·
80e9452
1
Parent(s):
d6429ec
Seed update
Browse files
app.py
CHANGED
@@ -583,12 +583,12 @@ class InferenceWrapper:
|
|
583 |
self.style_gan = StyleGAN(start_res=START_RES, target_res=TARGET_RES)
|
584 |
self.style_gan.grow_model(128)
|
585 |
self.style_gan.load_weights(os.path.join("pretrained/stylegan_128x128.ckpt"))
|
586 |
-
self.seed =
|
587 |
|
588 |
def __call__(self, seed, feature):
|
589 |
if seed != self.seed:
|
590 |
print(f"Loading model: {self.model}")
|
591 |
-
tf.random.set_seed(
|
592 |
batch_size = 1
|
593 |
self.z = tf.random.normal((batch_size, self.style_gan.z_dim))
|
594 |
self.w = self.style_gan.mapping(self.z)
|
|
|
583 |
self.style_gan = StyleGAN(start_res=START_RES, target_res=TARGET_RES)
|
584 |
self.style_gan.grow_model(128)
|
585 |
self.style_gan.load_weights(os.path.join("pretrained/stylegan_128x128.ckpt"))
|
586 |
+
self.seed = -1
|
587 |
|
588 |
def __call__(self, seed, feature):
|
589 |
if seed != self.seed:
|
590 |
print(f"Loading model: {self.model}")
|
591 |
+
tf.random.set_seed(seed)
|
592 |
batch_size = 1
|
593 |
self.z = tf.random.normal((batch_size, self.style_gan.z_dim))
|
594 |
self.w = self.style_gan.mapping(self.z)
|