Skip to content
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .ci/scripts/test_huggingface_optimum_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -369,7 +369,7 @@ def test_vit(model_id, model_dir, recipe, *, quantize=False, run_only=False):
), # fails to lower for CoreML
"smollm2-135m": ("HuggingFaceTB/SmolLM2-135M", test_text_generation),
"smollm3-3b": ("HuggingFaceTB/SmolLM3-3B", test_text_generation),
"olmo": ("allenai/OLMo-1B-hf", test_text_generation),
"olmo-1b": ("allenai/OLMo-1B-hf", test_text_generation),
}

_mask_fill_mapping = {
Expand Down
Loading