From bbd7c6e0972b36fd65c4b538e867bdc034b30dab Mon Sep 17 00:00:00 2001 From: Saquib Alam Date: Fri, 4 Aug 2023 03:16:38 +0530 Subject: [PATCH] added vae scaling factor --- comfy_extras/nodes_latent.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/comfy_extras/nodes_latent.py b/comfy_extras/nodes_latent.py index 8f2ee40df..abf82db1f 100644 --- a/comfy_extras/nodes_latent.py +++ b/comfy_extras/nodes_latent.py @@ -1,6 +1,9 @@ import torch from nodes import MAX_RESOLUTION +# diffusers library scale the random noise +vae_scaling_factor = 0.18215 + class NoisyLatentImage: def __init__(self, device="cpu"): self.device = device @@ -19,7 +22,7 @@ class NoisyLatentImage: def generate(self, seed, width, height, batch_size=1): 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}, )