nodes/city96-ComfyUI_ExtraModels
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Quick fix for new comfy version

Browse Source
This commit is contained in:
City
2024-04-06 12:05:04 +02:00
parent 6f80adfce5
commit 18fb3f5674
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
PixArt/sampler.py
+1 -3
View File
@@ -2,7 +2,7 @@ import torch
from .sampling import gaussian_diffusion as gd from .sampling import gaussian_diffusion as gd
from .sampling.dpm_solver import model_wrapper, DPM_Solver, NoiseScheduleVP from .sampling.dpm_solver import model_wrapper, DPM_Solver, NoiseScheduleVP
from comfy.sample import prepare_sampling, prepare_noise, cleanup_additional_models, get_models_from_cond from comfy.sample import prepare_sampling, prepare_noise
import comfy.utils import comfy.utils
import latent_preview import latent_preview
@@ -69,6 +69,4 @@ def sample_pixart(model, seed, steps, cfg, noise_schedule, noise_schedule_vp, po
previewer=previewer, previewer=previewer,
) )
cleanup_additional_models(models)
cleanup_additional_models(set(get_models_from_cond(positive, "control")))
return samples.detach().cpu().float() * (1 / model.model.latent_format.scale_factor) return samples.detach().cpu().float() * (1 / model.model.latent_format.scale_factor)