diff --git a/tests/generate_tests.py b/tests/generate_tests.py index 2e81645..adf35b3 100644 --- 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": [