Skip to content

Commit c944a59

Browse files
committed
Updated deprecated_api test to load model again before another export
Signed-off-by: Asmita Goswami <[email protected]>
1 parent a56879e commit c944a59

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/transformers/models/test_causal_lm_models.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -282,6 +282,10 @@ def test_causal_lm_export_with_deprecated_api(model_name):
282282
tokenizer = load_hf_tokenizer(pretrained_model_name_or_path=model_name)
283283
qeff_model = QEFFAutoModelForCausalLM(model, model_name=model_name, pretrained_model_name_or_path=model_name)
284284
new_api_onnx_model_path = qeff_model.export()
285+
286+
# Again loading model since the export moves model to meta device
287+
model, _ = load_causal_lm_model(model_name, n_layer=1)
288+
qeff_model = QEFFAutoModelForCausalLM(model, model_name=model_name, pretrained_model_name_or_path=model_name)
285289
_, old_api_onnx_model_path = qualcomm_efficient_converter(
286290
model_name=model_name, model_kv=qeff_model, tokenizer=tokenizer
287291
)

0 commit comments

Comments
 (0)