change default attention

2024-06-20 09:40:39 +03:00
parent 0d15baca15
commit eaa39a178d
1 changed files with 1 additions and 1 deletions
@@ -51,7 +51,7 @@ class DownloadAndLoadFlorence2Model:
            "attention": (
                    [ 'flash_attention_2', 'sdpa', 'eager'],
                    {
-                    "default": 'flash_attention_2'
+                    "default": 'sdpa'
                    }),

            },