File tree Expand file tree Collapse file tree 2 files changed +4
-5
lines changed Expand file tree Collapse file tree 2 files changed +4
-5
lines changed Original file line number Diff line number Diff line change 44load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
55load(":targets.bzl", "define_common_targets")
66
7+ # @noautodeps
78oncall("ai_infra_mobile_platform")
89
910define_common_targets()
Original file line number Diff line number Diff line change 1010"""
1111
1212import unittest
13+ from tempfile import TemporaryDirectory
14+
1315import pytest
1416from pytorch_tokenizers import CppHFTokenizer
1517from transformers import AutoTokenizer
16- from tempfile import TemporaryDirectory
1718
1819PROMPT = "What is the capital of France?"
1920
2021
21- @pytest .mark .parametrize ("model_id" , [
22- "HuggingFaceTB/SmolLM3-3B" ,
23- "Qwen/Qwen2.5-0.5B"
24- ])
22+ @pytest .mark .parametrize ("model_id" , ["HuggingFaceTB/SmolLM3-3B" , "Qwen/Qwen2.5-0.5B" ])
2523def test_models (model_id : str ) -> None :
2624 with TemporaryDirectory () as temp_dir :
2725 tokenizer = AutoTokenizer .from_pretrained (model_id )
You can’t perform that action at this time.
0 commit comments