ignore in test-tokenizers

e80ed76c03d05d408b2256fa67f2c1878b23e72e · 2024-11-06 22:16:13 +0400 · Alex Cheema

Files touched

M test/test_tokenizers.py

Diff

commit e80ed76c03d05d408b2256fa67f2c1878b23e72e
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Wed Nov 6 22:16:13 2024 +0400

    ignore in test-tokenizers
---
 test/test_tokenizers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/test_tokenizers.py b/test/test_tokenizers.py
index ec4d047b..820d4184 100644
--- a/test/test_tokenizers.py
+++ b/test/test_tokenizers.py
@@ -24,7 +24,7 @@ def test_tokenizer(name, tokenizer, verbose=False):
     strip_tokens = lambda s: s.lstrip(tokenizer.decode([tokenizer.bos_token_id])).rstrip(tokenizer.decode([tokenizer.eos_token_id]))
     assert text == strip_tokens(decoded) == strip_tokens(reconstructed)
 
-ignore = ["TriAiExperiments/SFR-Iterative-DPO-LLaMA-3-70B-R", "mlx-community/DeepSeek-Coder-V2-Lite-Instruct-4bit-mlx", "mlx-community/DeepSeek-V2.5-MLX-AQ4_1_64", "llava-hf/llava-1.5-7b-hf", "mlx-community/Qwen*", "dummy"]
+ignore = ["TriAiExperiments/SFR-Iterative-DPO-LLaMA-3-70B-R", "mlx-community/DeepSeek-Coder-V2-Lite-Instruct-4bit-mlx", "mlx-community/DeepSeek-V2.5-MLX-AQ4_1_64", "llava-hf/llava-1.5-7b-hf", "mlx-community/Qwen*", "dummy", "IntuitIntel/*"]
 ignore_pattern = re.compile(r"^(" + "|".join(model.replace("*", ".*") for model in ignore) + r")")
 models = [shard.model_id for shards in model_base_shards.values() for shard in shards.values() if not ignore_pattern.match(shard.model_id)]

← 36c1f68c update llama-3.1-405b-8bit model id to IntuitIntel/Meta-Llam · back to Exo · added new model info for 1B and 3B model sizes 029dc5f8 →