remove device definition

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