Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
d75a5b46cc | ||
|
|
1b6a016f78 |
@@ -118,6 +118,9 @@ jobs:
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
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' \
|
curl --silent --show-error --fail-with-body --location --request POST 'https://modelhub.org.cn//adminApi/image-verify' \
|
||||||
--header "Xc-Token: ${FIXED_TOKEN}" \
|
--header "Xc-Token: ${FIXED_TOKEN}" \
|
||||||
--header 'Content-Type: application/json' \
|
--header 'Content-Type: application/json' \
|
||||||
@@ -126,7 +129,6 @@ jobs:
|
|||||||
\"gpuType\": \"${GPU_TYPE}\",
|
\"gpuType\": \"${GPU_TYPE}\",
|
||||||
\"imageUrl\": \"${IMAGE}\",
|
\"imageUrl\": \"${IMAGE}\",
|
||||||
\"taskType\": \"${TASK_TYPE}\",
|
\"taskType\": \"${TASK_TYPE}\",
|
||||||
\"createBy\": \"${{ gitea.actor }}\",
|
\"createBy\": \"${{ gitea.event.pusher.email }}\",
|
||||||
\"repoUrl\": \"${{ gitea.server_url }}/${{ gitea.repository }}\",
|
\"repoUrl\": \"${{ gitea.server_url }}/${{ gitea.repository }}\"
|
||||||
\"tag\": \"${{ github.ref_name }}\"
|
|
||||||
}"
|
}"
|
||||||
|
|||||||
@@ -79,7 +79,7 @@ def load_model():
|
|||||||
processor_class = resolve_transformers_class(config["processer_class"])
|
processor_class = resolve_transformers_class(config["processer_class"])
|
||||||
model_class = resolve_transformers_class(config["model_class"])
|
model_class = resolve_transformers_class(config["model_class"])
|
||||||
torch_dtype = resolve_torch_dtype(config["torch_dtype"])
|
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(
|
logger.info(
|
||||||
f"model config: model_class={config['model_class']}, "
|
f"model config: model_class={config['model_class']}, "
|
||||||
|
|||||||
Reference in New Issue
Block a user