From a231eaa297464f19be44dcb22bdf95b97519ce03 Mon Sep 17 00:00:00 2001 From: Blake Date: Mon, 8 Apr 2024 15:29:40 -0700 Subject: [PATCH] Resolve the 'SDXL' object has no attribute 'get_model_object' with BNK_GetSigma --- nodes.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/nodes.py b/nodes.py index 2388578..56e05c7 100644 --- a/nodes.py +++ b/nodes.py @@ -137,10 +137,7 @@ def calc_sigma(self, model, sampler_name, scheduler, steps, start_at_step, end_a device = comfy.model_management.get_torch_device() end_at_step = min(steps, end_at_step) start_at_step = min(start_at_step, end_at_step) - real_model = None - comfy.model_management.load_model_gpu(model) - real_model = model.model - sampler = comfy.samplers.KSampler(real_model, steps=steps, device=device, sampler=sampler_name, scheduler=scheduler, denoise=1.0, model_options=model.model_options) + sampler = comfy.samplers.KSampler(model, steps=steps, device=device, sampler=sampler_name, scheduler=scheduler, denoise=1.0, model_options=model.model_options) sigmas = sampler.sigmas sigma = sigmas[start_at_step] - sigmas[end_at_step] sigma /= model.model.latent_format.scale_factor