Update docker-compose.yml
This commit is contained in:
parent
fce146ab68
commit
15608d0558
|
@ -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"
|
||||||
|
|
Loading…
Reference in New Issue