mirror of
https://github.com/corca-ai/EVAL
synced 2024-10-30 09:20:44 +00:00
23 lines
519 B
YAML
23 lines
519 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/
|
|
ports:
|
|
- "8000:8000"
|
|
env_file:
|
|
- .env
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
device_ids: ["3"] # You can choose which GPU to use
|
|
capabilities: [gpu]
|