File tree 4 files changed +18
-3
lines changed
4 files changed +18
-3
lines changed Original file line number Diff line number Diff line change @@ -100,13 +100,14 @@ test_model() {
100
100
rm " ./${MODEL_NAME} .pte"
101
101
return # Skip running with portable executor runnner since portable doesn't support Qwen's biased linears.
102
102
fi
103
- if [[ " ${MODEL_NAME} " == " phi4_mini " ]]; then
103
+ if [[ " ${MODEL_NAME} " == " phi-4-mini " ]]; then
104
104
# Install requirements for export_llama
105
105
bash examples/models/llama/install_requirements.sh
106
106
# Test export_llama script: python3 -m examples.models.llama.export_llama.
107
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
108
108
run_portable_executor_runner
109
109
rm " ./${MODEL_NAME} .pte"
110
+ return
110
111
fi
111
112
112
113
# Export a basic .pte and run the model.
Original file line number Diff line number Diff line change 35
35
"llava" : ("llava" , "LlavaModel" ),
36
36
"efficient_sam" : ("efficient_sam" , "EfficientSAM" ),
37
37
"qwen2_5" : ("qwen2_5" , "Qwen2_5Model" ),
38
- "phi4_mini " : ("phi4_mini " , "Phi4MiniModel" ),
38
+ "phi-4-mini " : ("phi-4-mini " , "Phi4MiniModel" ),
39
39
}
40
40
41
41
__all__ = [
Original file line number Diff line number Diff line change 93
93
"llama3_2" ,
94
94
"static_llama" ,
95
95
"qwen2_5" ,
96
- "phi4_mini " ,
96
+ "phi-4-mini " ,
97
97
]
98
98
TORCHTUNE_DEFINED_MODELS = ["llama3_2_vision" ]
99
99
Original file line number Diff line number Diff line change
1
+ # This source code is licensed under the BSD-style license found in the
2
+ # LICENSE file in the root directory of this source tree.
3
+
4
+ from executorch .examples .models .llama .model import Llama2Model
5
+
6
+
7
+ class Phi4MiniModel (Llama2Model ):
8
+ def __init__ (self , ** kwargs ):
9
+ super ().__init__ (** kwargs )
10
+
11
+
12
+ __all__ = [
13
+ "Phi4MiniModel" ,
14
+ ]
You can’t perform that action at this time.
0 commit comments