From e4a8fe201beef979a1b0d14be254778c781c72f2 Mon Sep 17 00:00:00 2001 From: Daniel King Date: Thu, 18 Apr 2024 11:36:04 -0700 Subject: [PATCH] fix --- tests/models/layers/test_huggingface_flash.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/models/layers/test_huggingface_flash.py b/tests/models/layers/test_huggingface_flash.py index e857fc0680..08891d5199 100644 --- 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, }