更新 transformers_server.py

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

@@ -118,9 +118,6 @@ jobs:
             exit 1
           fi
 
-          echo "gitea.actor=${{ gitea.actor }}"
-          echo "gitea.event.pusher.email=${{ gitea.event.pusher.email }}"
-
           curl --silent --show-error --fail-with-body --location --request POST 'https://modelhub.org.cn//adminApi/image-verify' \
             --header "Xc-Token: ${FIXED_TOKEN}" \
             --header 'Content-Type: application/json' \
@@ -129,6 +126,7 @@ jobs:
               \"gpuType\": \"${GPU_TYPE}\",
               \"imageUrl\": \"${IMAGE}\",
               \"taskType\": \"${TASK_TYPE}\",
-              \"createBy\": \"${{ gitea.event.pusher.email }}\",
-              \"repoUrl\": \"${{ gitea.server_url }}/${{ gitea.repository }}\"
+              \"createBy\": \"${{ gitea.actor }}\",
+              \"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']}, "