From b932036af30e05c06c82e29ed6a4045176248334 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Fri, 25 Oct 2024 22:37:08 +0300 Subject: [PATCH] Update asymm_models_joint.py --- mochi_preview/dit/joint_model/asymm_models_joint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mochi_preview/dit/joint_model/asymm_models_joint.py b/mochi_preview/dit/joint_model/asymm_models_joint.py index 68b49b0..ba0d67d 100644 --- a/mochi_preview/dit/joint_model/asymm_models_joint.py +++ b/mochi_preview/dit/joint_model/asymm_models_joint.py @@ -43,7 +43,7 @@ except ImportError: SAGEATTN_IS_AVAILABLE = False backends = [] -if torch.cuda.get_device_properties(0).major <= 7.5: +if torch.cuda.get_device_properties(0).major < 7: backends.append(SDPBackend.MATH) if torch.cuda.get_device_properties(0).major >= 9.0: backends.append(SDPBackend.CUDNN_ATTENTION)