Update docker-compose.yml
This commit is contained in:
parent
fce146ab68
commit
15608d0558
|
@ -1,8 +1,8 @@
|
|||
services:
|
||||
llamafactory:
|
||||
build:
|
||||
dockerfile: Dockerfile
|
||||
context: .
|
||||
dockerfile: ./docker/docker-cuda/Dockerfile
|
||||
context: ../..
|
||||
args:
|
||||
INSTALL_BNB: false
|
||||
INSTALL_VLLM: false
|
||||
|
@ -10,9 +10,9 @@ services:
|
|||
PIP_INDEX: https://pypi.org/simple
|
||||
container_name: llamafactory
|
||||
volumes:
|
||||
- ../../hf_cache:/root/.cache/huggingface/
|
||||
- ../../data:/app/LLaMA-Factory/data
|
||||
- ../../output:/app/LLaMA-Factory/output
|
||||
- ./hf_cache:/root/.cache/huggingface/
|
||||
- ./data:/app/data
|
||||
- ./output:/app/output
|
||||
ports:
|
||||
- "7860:7860"
|
||||
- "8000:8000"
|
||||
|
|
Loading…
Reference in New Issue