Update generate_tests.py

2023-10-03 11:52:58 +02:00 · 2023-10-03 11:52:58 +02:00 · 965a7154c5
parent 5b31129218
commit 965a7154c5
1 changed files with 1 additions and 1 deletions
--- a/tests/generate_tests.py
+++ b/tests/generate_tests.py
@ -10,7 +10,7 @@ from scripts.supported_models import SUPPORTED_MODELS

 # List of tokenizers where the model isn't yet supported, but the tokenizer is
 ADDITIONAL_TOKENIZERS_TO_TEST = {
-    'RefinedWebModel': [
+    'falcon': [
        'tiiuae/falcon-7b',
    ],
    "llama": [