From e28962199273d72d691a3764cb0377969123dd42 Mon Sep 17 00:00:00 2001
From: Stephan Auerhahn <stephan@stability.ai>
Date: Sat, 12 Aug 2023 13:52:46 -0700
Subject: [PATCH] fix reference

---
 sgm/inference/api.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sgm/inference/api.py b/sgm/inference/api.py
index 87592dc..f4f2faa 100644
--- a/sgm/inference/api.py
+++ b/sgm/inference/api.py
@@ -394,7 +394,7 @@ class SamplingPipeline:
             "negative_aesthetic_score": 2.5,
         }
 
-        sampler.discretization = self.wrap_discretization(
+        sampler.discretization = wrap_discretization(
             sampler.discretization, image_strength=params.img2img_strength
         )