Update docker-compose.yml

Former-commit-id: e038daf8dfa5d948b70c18469cb5a0be9aec464a
This commit is contained in:
hoshi-hiyouga 2024-06-25 00:54:28 +08:00 committed by GitHub
parent 37013f8068
commit 8367ec03a7

View File

@ -1,20 +1,19 @@
services: services:
llamafactory: llamafactory:
build: build:
dockerfile: Dockerfile dockerfile: ./docker/docker-npu/Dockerfile
context: . context: ../..
args: args:
INSTALL_DEEPSPEED: false INSTALL_DEEPSPEED: false
PIP_INDEX: https://pypi.org/simple PIP_INDEX: https://pypi.org/simple
container_name: llamafactory container_name: llamafactory
volumes: volumes:
- ../../hf_cache:/root/.cache/huggingface/ - ./hf_cache:/root/.cache/huggingface/
- ../../data:/app/LLaMA-Factory/data - ./data:/app/data
- ../../output:/app/LLaMA-Factory/output - ./output:/app/output
- /usr/local/dcmi:/usr/local/dcmi - /usr/local/dcmi:/usr/local/dcmi
- /usr/local/bin/npu-smi:/usr/local/bin/npu-smi - /usr/local/bin/npu-smi:/usr/local/bin/npu-smi
- /usr/local/Ascend/driver/lib64:/usr/local/Ascend/driver/lib64 - /usr/local/Ascend/driver:/usr/local/Ascend/driver
- /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info
- /etc/ascend_install.info:/etc/ascend_install.info - /etc/ascend_install.info:/etc/ascend_install.info
ports: ports:
- "7860:7860" - "7860:7860"