mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-06-22 15:59:45 +08:00
Update nodes_latent.py
This commit is contained in:
parent
53798281a1
commit
6dbb89ac99
@ -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}, )
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user