更新 transformers_server.py

.gitea/workflows/docker-build-push.yml

@@ -127,5 +127,6 @@ jobs:
               \"imageUrl\": \"${IMAGE}\",
               \"taskType\": \"${TASK_TYPE}\",
               \"createBy\": \"${{ gitea.actor }}\",
-              \"repoUrl\": \"${{ gitea.server_url }}/${{ gitea.repository }}\"
+              \"repoUrl\": \"${{ gitea.server_url }}/${{ gitea.repository }}\",
+              \"tag\": \"${{ github.ref_name }}\"
             }"

transformers_server.py

@@ -79,7 +79,7 @@ def load_model():
     processor_class = resolve_transformers_class(config["processer_class"])
     model_class = resolve_transformers_class(config["model_class"])
     torch_dtype = resolve_torch_dtype(config["torch_dtype"])
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    device = torch.device("cuda")
 
     logger.info(
         f"model config: model_class={config['model_class']}, "