mirror of
https://git.isriupjv.fr/ISRI/ai-server
synced 2025-04-24 10:08:11 +02:00
17 lines
331 B
YAML
17 lines
331 B
YAML
services:
|
|
ai-server:
|
|
build:
|
|
context: .
|
|
dockerfile: ./Dockerfile
|
|
runtime: nvidia
|
|
volumes:
|
|
- models:/models/
|
|
- /root/.cache/huggingface:/root/.cache/huggingface
|
|
environment:
|
|
- MODEL_LIBRARY=/models/
|
|
- NVIDIA_VISIBLE_DEVICES=all
|
|
ports:
|
|
- "8000:8000"
|
|
|
|
volumes:
|
|
models:
|