diff --git a/ldm_patched/contrib/external_model_downscale.py b/ldm_patched/contrib/external_model_downscale.py index 58a01d55..8a3d67cd 100644 --- a/ldm_patched/contrib/external_model_downscale.py +++ b/ldm_patched/contrib/external_model_downscale.py @@ -24,8 +24,8 @@ class PatchModelAddDownscale: CATEGORY = "_for_testing" def patch(self, model, block_number, downscale_factor, start_percent, end_percent, downscale_after_skip, downscale_method, upscale_method): - sigma_start = model.model.model_sampling.percent_to_sigma(start_percent) - sigma_end = model.model.model_sampling.percent_to_sigma(end_percent) + sigma_start = model.model.predictor.percent_to_sigma(start_percent) + sigma_end = model.model.predictor.percent_to_sigma(end_percent) def input_block_patch(h, transformer_options): if transformer_options["block"][1] == block_number: