Skip to content

Commit 2f3ea4a

Browse files
committed
Rename phi-4-mini to phi_4_mini
1 parent 8cd1b93 commit 2f3ea4a

File tree

6 files changed

+5
-5
lines changed

6 files changed

+5
-5
lines changed

.ci/scripts/test_model.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,11 +100,11 @@ test_model() {
100100
rm "./${MODEL_NAME}.pte"
101101
return # Skip running with portable executor runnner since portable doesn't support Qwen's biased linears.
102102
fi
103-
if [[ "${MODEL_NAME}" == "phi-4-mini" ]]; then
103+
if [[ "${MODEL_NAME}" == "phi_4_mini" ]]; then
104104
# Install requirements for export_llama
105105
bash examples/models/llama/install_requirements.sh
106106
# Test export_llama script: python3 -m examples.models.llama.export_llama.
107-
"${PYTHON_EXECUTABLE}" -m examples.models.llama.export_llama --model "${MODEL_NAME}" -c examples/models/llama/params/demo_rand_params.pth -p examples/models/phi-4-mini/config.json
107+
"${PYTHON_EXECUTABLE}" -m examples.models.llama.export_llama --model "${MODEL_NAME}" -c examples/models/llama/params/demo_rand_params.pth -p examples/models/phi_4_mini/config.json
108108
run_portable_executor_runner
109109
rm "./${MODEL_NAME}.pte"
110110
return

examples/models/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ class Model(str, Enum):
3636
Llava = "llava"
3737
EfficientSam = "efficient_sam"
3838
Qwen25 = "qwen2_5"
39-
Phi4Mini = "phi-4-mini"
39+
Phi4Mini = "phi_4_mini"
4040

4141
def __str__(self) -> str:
4242
return self.value
@@ -80,7 +80,7 @@ def __str__(self) -> str:
8080
str(Model.Llava): ("llava", "LlavaModel"),
8181
str(Model.EfficientSam): ("efficient_sam", "EfficientSAM"),
8282
str(Model.Qwen25): ("qwen2_5", "Qwen2_5Model"),
83-
str(Model.Phi4Mini): ("phi-4-mini", "Phi4MiniModel"),
83+
str(Model.Phi4Mini): ("phi_4_mini", "Phi4MiniModel"),
8484
}
8585

8686
__all__ = [

examples/models/llama/export_llama_lib.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@
9595
"llama3_2",
9696
"static_llama",
9797
"qwen2_5",
98-
"phi-4-mini",
98+
"phi_4_mini",
9999
]
100100
TORCHTUNE_DEFINED_MODELS = ["llama3_2_vision"]
101101

0 commit comments

Comments
 (0)