Update nodes_latent.py

This commit is contained in:
Saquib Alam 2023-08-04 04:29:56 +05:30 committed by GitHub
parent 53798281a1
commit 6dbb89ac99
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -22,7 +22,7 @@ class NoisyLatentImage:
def generate(self, seed, width, height, batch_size=1): def generate(self, seed, width, height, batch_size=1):
generator = torch.manual_seed(seed) generator = torch.manual_seed(seed)
latent = torch.randn([batch_size, 4, height // 8, width // 8], generator=generator, device=self.device) latent = torch.randn([batch_size, 4, height // 8, width // 8], generator=generator, device=self.device) / vae_scaling_factor
return ({"samples":latent}, ) return ({"samples":latent}, )