From cb588875b4fa14509d8a9341dbcb8b6f134223b1 Mon Sep 17 00:00:00 2001 From: Pam Date: Mon, 6 May 2024 01:01:13 +0500 Subject: [PATCH] Use get_model_object instead of model.model_sampling --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index 0ffdc29..c98ce30 100644 --- a/nodes.py +++ b/nodes.py @@ -88,7 +88,7 @@ def get_denoised_ranges(latent, measure="hard", top_k=0.25): return chans def get_sigmin_sigmax(model): - model_sampling = model.model.model_sampling + model_sampling = model.get_model_object("model_sampling") sigmin = model_sampling.sigma(model_sampling.timestep(model_sampling.sigma_min)) sigmax = model_sampling.sigma(model_sampling.timestep(model_sampling.sigma_max)) return sigmin, sigmax