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

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

@@ -130,12 +130,15 @@ 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.to(device)
+            g_audio_encoder.cuda(device)
             g_audio_encoder.eval()
     else:
-        model.to(device).eval()
+        #model.to(device).eval()
+        model.cuda(device).eval()
         if args.model_type == "stf_v3":
-            g_audio_encoder.to(device).eval()
+            #g_audio_encoder.to(device).eval()
+            g_audio_encoder.cuda(device).eval()
 
     model_data = ModelInfo(
         model=model,