Commit 65a87ccc authored by Kohaku-Blueleaf's avatar Kohaku-Blueleaf
Browse files

Add error information for recursion error

parent 302d95c7
Loading
Loading
Loading
Loading
+6 −0
Original line number Diff line number Diff line
@@ -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