diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-23 16:40:20 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-23 16:40:20 +0300 |
| commit | 59146621e256269b85feb536edeb745da20daf68 (patch) | |
| tree | 4210d6c2d709be8472dcb3584afb54e4284fd09f /modules/errors.py | |
| parent | 3fa482076a5f07d81d37d58a31b8c4fe3a740843 (diff) | |
better support for xformers flash attention on older versions of torch
Diffstat (limited to 'modules/errors.py')
| -rw-r--r-- | modules/errors.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/modules/errors.py b/modules/errors.py index a10e8708..f6b80dbb 100644 --- a/modules/errors.py +++ b/modules/errors.py @@ -24,6 +24,18 @@ See https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Features#stable """)
+already_displayed = {}
+
+
+def display_once(e: Exception, task):
+ if task in already_displayed:
+ return
+
+ display(e, task)
+
+ already_displayed[task] = 1
+
+
def run(code, task):
try:
code()
|