Add error information for recursion error
This commit is contained in:
parent
302d95c726
commit
65a87ccc9b
@ -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
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user