Update docker-compose.yml

This commit is contained in:
hoshi-hiyouga 2024-06-25 00:46:47 +08:00 committed by GitHub
parent fce146ab68
commit 15608d0558
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions

View File

@ -1,8 +1,8 @@
services: services:
llamafactory: llamafactory:
build: build:
dockerfile: Dockerfile dockerfile: ./docker/docker-cuda/Dockerfile
context: . context: ../..
args: args:
INSTALL_BNB: false INSTALL_BNB: false
INSTALL_VLLM: false INSTALL_VLLM: false
@ -10,9 +10,9 @@ services:
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
ports: ports:
- "7860:7860" - "7860:7860"
- "8000:8000" - "8000:8000"