moflo commited on
Commit
80e9452
1 Parent(s): d6429ec

Seed update

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = 196
587
 
588
  def __call__(self, seed, feature):
589
  if seed != self.seed:
590
  print(f"Loading model: {self.model}")
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)
 
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)