up

run_demo_openai.py

@@ -36,7 +36,9 @@ GEN_KWARGS = {
     # "no_speech_threshold": None,
 }
 
-device = 0 if torch.cuda.is_available() else "cpu"
+# device = 0 if torch.cuda.is_available() else "cpu"
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+
 
 downloaded_model_path = hf_hub_download(repo_id=MODEL_NAME, filename=CHECKPOINT_FILENAME)