xinyun/ComfyUI-MochiWrapper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

backends

Browse Source
This commit is contained in:
kijai 2024-10-26 17:49:15 +03:00
parent f29f739707
commit ddfb3a6bf2
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mochi_preview/dit/joint_model/asymm_models_joint.py
View File

@ -46,7 +46,9 @@ except ImportError:
backends = []
if torch.cuda.get_device_properties(0).major < 7:
backends.append(SDPBackend.MATH)
backends.append(SDPBackend.EFFICIENT_ATTENTION)
if torch.cuda.get_device_properties(0).major >= 9.0:
backends.append(SDPBackend.EFFICIENT_ATTENTION)
backends.append(SDPBackend.CUDNN_ATTENTION)
else:
backends.append(SDPBackend.EFFICIENT_ATTENTION)