jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

made sample functions more explicit

Browse Source
This commit is contained in:
BlenderNeko
2023-04-24 12:53:10 +02:00
parent 5818539743
commit d9b1595f85
2 changed files with 29 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
nodes.py
View File

@@ -747,9 +747,12 @@ def common_ksampler(model, seed, steps, cfg, sampler_name, scheduler, positive,
if disable_noise:
noise = torch.zeros(latent_image.size(), dtype=latent_image.dtype, layout=latent_image.layout, device="cpu")
else:
noise = comfy.sample.prepare_noise(latent, seed)
skip = latent["batch_index"] if "batch_index" in latent else 0
noise = comfy.sample.prepare_noise(latent_image, seed, skip)
noise_mask = comfy.sample.create_mask(latent, noise)
noise_mask = None
if "noise_mask" in latent:
noise_mask = comfy.sample.prepare_mask(latent["noise_mask"], noise)
real_model = None
comfy.model_management.load_model_gpu(model)