diff --git a/libs/infinity_emb/Docker.template.yaml b/libs/infinity_emb/Docker.template.yaml index 95757c7d..d2e2e6af 100644 --- a/libs/infinity_emb/Docker.template.yaml +++ b/libs/infinity_emb/Docker.template.yaml @@ -66,8 +66,7 @@ amd: && cd /tmp-optimum \ && python -m pip install .; \ else \ - echo "Unsupported GPU_ARCH: ${GPU_ARCH}"; \ - exit(1); \ + echo "NO GPU_ARCH, skip optium: ${GPU_ARCH}"; \ fi poetry_extras: "all" python_version: python3.10