From 7735706b939847df2c6c00b44c36f95a20137c61 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jie=20Fu=20=28=E5=82=85=E6=9D=B0=29?= Date: Wed, 24 Sep 2025 14:46:52 +0800 Subject: [PATCH] model-conversion : run-org-model.py fails to run on mac m1 (#16213) Signed-off-by: Jie Fu --- examples/model-conversion/scripts/causal/run-org-model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/model-conversion/scripts/causal/run-org-model.py b/examples/model-conversion/scripts/causal/run-org-model.py index 78a54abf13..9444c713d0 100755 --- a/examples/model-conversion/scripts/causal/run-org-model.py +++ b/examples/model-conversion/scripts/causal/run-org-model.py @@ -193,7 +193,7 @@ print(f"Input text: {repr(prompt)}") print(f"Tokenized: {tokenizer.convert_ids_to_tokens(input_ids[0])}") with torch.no_grad(): - outputs = model(input_ids) + outputs = model(input_ids.to(model.device)) logits = outputs.logits # Extract logits for the last token (next token prediction)