fix

mosaicml · Apr 18, 2024 · e4a8fe2 · e4a8fe2
1 parent 5038e27
commit e4a8fe2
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/models/layers/test_huggingface_flash.py b/tests/models/layers/test_huggingface_flash.py
@@ -27,7 +27,6 @@ def test_flash2(model_name: str, use_flash_attention_2: bool, init_device: str):
                 'intermediate_size': 64,
                 'hidden_size': 64,
             },
-            'use_auth_token': True,
             'pretrained': False,
             'init_device': init_device,
         }