Chaerin5 commited on
Commit
0ae1eb4
·
1 Parent(s): 785832a

fix vae nan bug

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -490,6 +490,8 @@ def get_ref_anno(ref):
490
  )
491
  print(f"image.max(): {image.max()}, image.min(): {image.min()}")
492
  print(f"opts.latent_scaling_factor: {opts.latent_scaling_factor}")
 
 
493
  latent = opts.latent_scaling_factor * autoencoder.encode(image).sample()
494
  print(f"latent.max(): {latent.max()}, latent.min(): {latent.min()}")
495
  if not REF_POSE_MASK:
 
490
  )
491
  print(f"image.max(): {image.max()}, image.min(): {image.min()}")
492
  print(f"opts.latent_scaling_factor: {opts.latent_scaling_factor}")
493
+ print(f"autoencoder encoder before operating max: {min([p.min() for p in autoencoder.encoder.parameters()])}")
494
+ print(f"autoencoder encoder before operating min: {max([p.max() for p in autoencoder.encoder.parameters()])}")
495
  latent = opts.latent_scaling_factor * autoencoder.encode(image).sample()
496
  print(f"latent.max(): {latent.max()}, latent.min(): {latent.min()}")
497
  if not REF_POSE_MASK: