Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
ComfyUI
Commits
2aed53c4
Commit
2aed53c4
authored
Apr 30, 2024
by
comfyanonymous
Browse files
Workaround xformers bug.
parent
bacce529
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
comfy/ldm/modules/attention.py
comfy/ldm/modules/attention.py
+2
-2
No files found.
comfy/ldm/modules/attention.py
View file @
2aed53c4
...
...
@@ -297,8 +297,8 @@ def attention_split(q, k, v, heads, mask=None):
BROKEN_XFORMERS
=
False
try
:
x_vers
=
xformers
.
__version__
#
I think 0.0.23 is also broken
(q with bs bigger than 65535 gives CUDA error)
BROKEN_XFORMERS
=
x_vers
.
startswith
(
"0.0.2
1
"
)
or
x_vers
.
startswith
(
"0.0.22"
)
or
x_vers
.
startswith
(
"0.0.2
3
"
)
#
XFormers bug confirmed on all versions from 0.0.21 to 0.0.26
(q with bs bigger than 65535 gives CUDA error)
BROKEN_XFORMERS
=
x_vers
.
startswith
(
"0.0.2"
)
and
not
x_vers
.
startswith
(
"0.0.2
0
"
)
except
:
pass
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment