From 018903ce2a050cf5d5579c87d7b6938e58a47314 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Fri, 2 Aug 2024 00:37:14 -0700
Subject: [PATCH] fix kohya-hr

---
 ldm_patched/contrib/external_model_downscale.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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: