Revert

EvolvingLMMs-Lab · Apr 5, 2024 · c4955df · c4955df
1 parent 86732eb
commit c4955df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lmms_eval/models/llava.py b/lmms_eval/models/llava.py
@@ -52,7 +52,7 @@ def __init__(
         batch_size: Optional[Union[int, str]] = 1,
         trust_remote_code: Optional[bool] = False,
         revision=None,
-        use_flash_attention_2=False, # True,
+        use_flash_attention_2=True,
         device_map="",
         conv_template="vicuna_v1",
         use_cache=True,