Commit
·
3b328e0
1
Parent(s):
8fba3f2
false
Browse files
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=
|
| 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)
|