ccclemenfff commited on
Commit
3b328e0
·
1 Parent(s): 8fba3f2
robohusky/model/configuration_husky.py CHANGED
@@ -196,7 +196,7 @@ class HuskyQFormerConfig(PretrainedConfig):
196
  classifier_dropout=None,
197
  cross_attention_frequency=2,
198
  encoder_hidden_size=1408,
199
- _flash_attn_2_enabled=True,
200
  **kwargs,
201
  ):
202
  super().__init__(pad_token_id=pad_token_id, **kwargs)
 
196
  classifier_dropout=None,
197
  cross_attention_frequency=2,
198
  encoder_hidden_size=1408,
199
+ _flash_attn_2_enabled=False,
200
  **kwargs,
201
  ):
202
  super().__init__(pad_token_id=pad_token_id, **kwargs)