Merge pull request #16 from pamparamm/main

Use get_model_object instead of model.model_sampling
2024-05-07 18:17:50 +02:00
parent 952edb7754 cb588875b4
commit 0725f9a197
1 changed files with 1 additions and 1 deletions
@@ -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