Add error information for recursion error

This commit is contained in:
Kohaku-Blueleaf 2023-05-23 00:09:49 +08:00
parent 302d95c726
commit 65a87ccc9b

View File

@ -272,6 +272,12 @@ class KDiffusionSampler:
try: try:
return func() 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: except sd_samplers_common.InterruptedException:
return self.last_latent return self.last_latent