Spaces:
Runtime error
Runtime error
moflo
commited on
Commit
•
d6429ec
1
Parent(s):
445c03a
Missing z
Browse files
app.py
CHANGED
@@ -591,7 +591,7 @@ class InferenceWrapper:
|
|
591 |
tf.random.set_seed(196)
|
592 |
batch_size = 1
|
593 |
self.z = tf.random.normal((batch_size, self.style_gan.z_dim))
|
594 |
-
self.w = self.style_gan.mapping(z)
|
595 |
self.noise = self.style_gan.generate_noise(batch_size=batch_size)
|
596 |
else:
|
597 |
print(f"Model '{self.model}' already loaded, reusing it.")
|
|
|
591 |
tf.random.set_seed(196)
|
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)
|
595 |
self.noise = self.style_gan.generate_noise(batch_size=batch_size)
|
596 |
else:
|
597 |
print(f"Model '{self.model}' already loaded, reusing it.")
|