wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-23 00:09:32 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

added vae scaling factor

Browse Source
This commit is contained in:
Saquib Alam 2023-08-04 03:16:38 +05:30 committed by GitHub
parent df0442a521
commit bbd7c6e097
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy_extras/nodes_latent.py
View File

@ -1,6 +1,9 @@
import torch import torch
from nodes import MAX_RESOLUTION from nodes import MAX_RESOLUTION
# diffusers library scale the random noise
vae_scaling_factor = 0.18215
class NoisyLatentImage: class NoisyLatentImage:
def __init__(self, device="cpu"): def __init__(self, device="cpu"):
self.device = device self.device = device
@ -19,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}, )