Add error information for recursion error

65a87ccc · Kohaku-Blueleaf · 302d95c7 · 65a87ccc
隐藏空白更改
内联并排

Showing with 6 addition and 0 deletion

modules/sd_samplers_kdiffusion.py modules/sd_samplers_kdiffusion.py +6 -0

未找到文件。
--- a/modules/sd_samplers_kdiffusion.py
+++ b/modules/sd_samplers_kdiffusion.py
@@ -272,6 +272,12 @@ class KDiffusionSampler:

        try:
            return func()
+        except RecursionError:
+            print(
+                'rho>5 with polyexponential scheduler may cause this error.'
+                'You should try to use smaller rho instead.'
+            )
+            return self.last_latent
        except sd_samplers_common.InterruptedException:
            return self.last_latent