cicd: post verify api

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

@@ -26,57 +26,19 @@ jobs:
       - name: Load and Validate Task Info
         run: |
           set -a
-          . .gitea/workflows/task_info.env
+          . ./task_info.env
           set +a
 
           for name in FRAMEWORK GPU_TYPE TASK_TYPE; do
             eval "value=\${${name}:-}"
-            if [ "$name" = "FRAMEWORK" ] && [ -z "$value" ]; then
-              echo "${name} is empty in .gitea/workflows/task_info.env"
+            if [ -z "$value" ]; then
+              echo "${name} is empty in task_info.env"
               exit 1
             fi
 
             echo "${name}=${value}" >> "$GITEA_ENV"
           done
 
-      - name: Validate Image Verify Metadata
-        run: |
-          if [ -z "${FIXED_TOKEN:-}" ]; then
-            echo "FIXED_TOKEN is not configured on runner"
-            exit 1
-          fi
-
-          if ! response="$(curl --silent --show-error --location --get 'https://modelhub.org.cn/adminApi/image-verify/validate' \
-            --header "Xc-Token: ${FIXED_TOKEN}" \
-            --data-urlencode "gpuType=${GPU_TYPE:-}" \
-            --data-urlencode "taskType=${TASK_TYPE:-}")"; then
-            echo "failed to call image verify validate API"
-            exit 1
-          fi
-
-          VALIDATE_RESPONSE="$response" python3 - <<'PY'
-          import json
-          import os
-          import sys
-
-          raw = os.environ.get("VALIDATE_RESPONSE", "")
-          try:
-              body = json.loads(raw)
-          except json.JSONDecodeError:
-              print("image verify validate API returned invalid JSON")
-              print(raw)
-              sys.exit(1)
-
-          if body.get("code") == 0 and body.get("data") is True:
-              print("image verify metadata validation passed")
-              sys.exit(0)
-
-          message = body.get("message") or "unknown error"
-          print(f"image verify metadata validation failed: {message}")
-          print(raw)
-          sys.exit(1)
-          PY
-
       - name: Login to Docker Registry
         run: |
           echo "$DOCKER_PASSWORD" | docker login "$DOCKER_REGISTRY" \
@@ -118,15 +80,12 @@ jobs:
             exit 1
           fi
 
-          curl --silent --show-error --fail-with-body --location --request POST 'https://modelhub.org.cn//adminApi/image-verify' \
+          curl --fail-with-body --location --request POST 'https://modelhub.org.cn//adminApi/image-verify' \
             --header "Xc-Token: ${FIXED_TOKEN}" \
             --header 'Content-Type: application/json' \
             --data-raw "{
               \"framework\": \"${FRAMEWORK}\",
               \"gpuType\": \"${GPU_TYPE}\",
               \"imageUrl\": \"${IMAGE}\",
-              \"taskType\": \"${TASK_TYPE}\",
-              \"createBy\": \"${{ gitea.actor }}\",
-              \"repoUrl\": \"${{ gitea.server_url }}/${{ gitea.repository }}\",
-              \"tag\": \"${{ github.ref_name }}\"
+              \"taskType\": \"${TASK_TYPE}\"
             }"

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")
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     logger.info(
         f"model config: model_class={config['model_class']}, "