Fix the docker image
This commit is contained in:
hiyouga
2024-06-11 00:19:17 +08:00
parent 0012762b04
commit 949e9908ad
4 changed files with 78 additions and 40 deletions

View File

@@ -1,17 +1,23 @@
version: '3.8'
services:
llama-factory:
llamafactory:
build:
dockerfile: Dockerfile
context: .
container_name: llama_factory
args:
INSTALL_BNB: false
INSTALL_VLLM: false
INSTALL_DEEPSPEED: false
PIP_INDEX: https://pypi.org/simple
container_name: llamafactory
volumes:
- ./hf_cache:/root/.cache/huggingface/
- ./data:/app/data
- ./output:/app/output
ports:
- "7860:7860"
- "8000:8000"
ipc: host
deploy:
resources: