You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
EVAL/docker-compose.yml

26 lines
602 B
YAML

version: "3"
services:
eval:
container_name: eval
image: eval
build:
dockerfile: Dockerfile
context: .
volumes: # if you want to decrease your model download time, use this.
- ../.cache/huggingface/:/root/.cache/huggingface/
- ./static/:/app/static/
ports:
- "7500:7500"
- "7501:7501"
- "8000:8000" # eval port
env_file:
- .env
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: ["3"] # You can choose which GPU to use
capabilities: [gpu]