configuration_husky.py HuskyVisionConfig

- _flash_attn_2_enabled=True,
+ _flash_attn_2_enabled=False,

robohusky/model/configuration_husky.py

@@ -88,7 +88,7 @@ class HuskyVisionConfig(PretrainedConfig):
             initializer_range=1e-10,
             initializer_factor=1.0,
             qkv_bias=True,
-            _flash_attn_2_enabled=True,
+            _flash_attn_2_enabled=False,
             **kwargs,
     ):
         super().__init__(**kwargs)