更新 transformers_server.py

add tag to cicd
2026-05-07 17:22:34 +08:00 · 2026-05-07 12:29:57 +08:00
2 changed files with 3 additions and 2 deletions
--- a/.gitea/workflows/docker-build-push.yml
+++ b/.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 }}\"
            }"
--- a/transformers_server.py
+++ b/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']}, "
Author	SHA1	Message	Date
aiyueqi	baed0b5003	更新 transformers_server.py All checks were successful Docker Build and Push / docker (push) Successful in 2m24s Details	2026-05-07 17:22:34 +08:00
aiyueqi	2e2e127738	add tag to cicd All checks were successful Docker Build and Push / docker (push) Successful in 3m23s Details	2026-05-07 12:29:57 +08:00