File tree Expand file tree Collapse file tree 2 files changed +3
-4
lines changed Expand file tree Collapse file tree 2 files changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -91,11 +91,10 @@ test_model() {
91
91
run_portable_executor_runner
92
92
rm " ./${MODEL_NAME} .pte"
93
93
fi
94
- STRICT=" --strict"
94
+ STRICT=" --no- strict"
95
95
if [[ " ${MODEL_NAME} " == " llava" ]]; then
96
96
# Install requirements for llava
97
97
bash examples/models/llava/install_requirements.sh
98
- STRICT=" --no-strict"
99
98
fi
100
99
if [[ " ${MODEL_NAME} " == " qwen2_5" ]]; then
101
100
# Install requirements for export_llama
Original file line number Diff line number Diff line change 87
87
88
88
model = model .eval ()
89
89
# pre-autograd export. eventually this will become torch.export
90
- ep = torch .export .export_for_training (model , example_inputs , strict = True )
90
+ ep = torch .export .export_for_training (model , example_inputs , strict = False )
91
91
model = ep .module ()
92
92
93
93
if args .quantize :
94
94
logging .info ("Quantizing Model..." )
95
95
# TODO(T165162973): This pass shall eventually be folded into quantizer
96
96
model = quantize (model , example_inputs , quant_type )
97
- ep = torch .export .export_for_training (model , example_inputs , strict = True )
97
+ ep = torch .export .export_for_training (model , example_inputs , strict = False )
98
98
99
99
edge = to_edge_transform_and_lower (
100
100
ep ,
You can’t perform that action at this time.
0 commit comments