Update stf/stf-api-alternative/src/stf_alternative/model.py

stf/stf-api-alternative/src/stf_alternative/model.py

@@ -131,14 +131,14 @@ def create_model(
         if args.model_type == "stf_v3":
             g_audio_encoder = torch.nn.DataParallel(g_audio_encoder, device_ids=gpus)
             #g_audio_encoder.to(device)
-            g_audio_encoder.cuda(device)
+            g_audio_encoder.cuda(0)
             g_audio_encoder.eval()
     else:
         #model.to(device).eval()
-        model.cuda(device).eval()
+        model.cuda(0).eval()
         if args.model_type == "stf_v3":
             #g_audio_encoder.to(device).eval()
-            g_audio_encoder.cuda(device).eval()
+            g_audio_encoder.cuda(0).eval()
 
     model_data = ModelInfo(
         model=model,