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 4
4
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
5
5
load(":targets.bzl", "define_common_targets")
6
6
7
+ # @noautodeps
7
8
oncall("ai_infra_mobile_platform")
8
9
9
10
define_common_targets()
Original file line number Diff line number Diff line change 10
10
"""
11
11
12
12
import unittest
13
+ from tempfile import TemporaryDirectory
14
+
13
15
import pytest
14
16
from pytorch_tokenizers import CppHFTokenizer
15
17
from transformers import AutoTokenizer
16
- from tempfile import TemporaryDirectory
17
18
18
19
PROMPT = "What is the capital of France?"
19
20
20
21
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" ])
25
23
def test_models (model_id : str ) -> None :
26
24
with TemporaryDirectory () as temp_dir :
27
25
tokenizer = AutoTokenizer .from_pretrained (model_id )
You can’t perform that action at this time.
0 commit comments