Spaces:
Running
on
Zero
Running
on
Zero
fix vae nan bug
Browse files
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:
|