diff --git a/docker/.env b/docker/.env index 03950b5..837f3eb 100644 --- a/docker/.env +++ b/docker/.env @@ -15,6 +15,9 @@ OUTPUT_DIR=./data/output MODEL_DIR=./data/models MIOPEN_CACHE=./data/miopen +# ---- 代理设置(国内访问 GitHub 慢时开启)---- +# GIT_PROXY=http://127.0.0.1:8118 + # ---- 模型下载源 ---- # huggingface(默认,需科学上网)或 modelscope(国内可用) MINERU_MODEL_SOURCE=modelscope diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml index 52a2ef4..b143dea 100644 --- a/docker/docker-compose.yml +++ b/docker/docker-compose.yml @@ -41,8 +41,8 @@ services: dockerfile: Dockerfile network: host args: - ARCH: gfx1201 - GIT_PROXY: http://127.0.0.1:8118 + ARCH: ${ARCH:-gfx1201} + GIT_PROXY: ${GIT_PROXY:-} container_name: mineru-worker0 stdin_open: true tty: true