move to device

huggingface · Mar 6, 2025 · 0b18c67 · 0b18c67
1 parent cd7d2c7
commit 0b18c67
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/question-answering/run_qa.py b/examples/question-answering/run_qa.py
@@ -376,6 +376,8 @@ def main():
         token=model_args.token,
         trust_remote_code=model_args.trust_remote_code,
     )
+    device = torch.device("hpu")
+    model = model.to(device)
 
     # Tokenizer check: this script requires a fast tokenizer.
     if not isinstance(tokenizer, PreTrainedTokenizerFast):