diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml index 11e3d18..8684f02 100644 --- a/docker/docker-compose.yml +++ b/docker/docker-compose.yml @@ -104,9 +104,12 @@ services: - "8000:8000" environment: - MINERU_MODEL_SOURCE=${MINERU_MODEL_SOURCE:-huggingface} + - HF_HUB_CACHE=${HF_HUB_CACHE:-/opt/models/huggingface} + - MODELSCOPE_CACHE=${MODELSCOPE_CACHE:-/opt/models/modelscope} volumes: - ${INPUT_DIR:-./data/input}:/data/input:ro - ${OUTPUT_DIR:-./data/output}:/data/output + - ${MODEL_DIR:-./data/models}:/opt/models - ./scripts:/opt/scripts:ro command: [