mdipierro
/
SmolDocling-256M-Demo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix compose

This commit is contained in:
Maurizio Dipierro 2025-03-19 16:23:45 +01:00
parent 671f475e11
commit be8aaff73c
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
docker-compose.yml
View File

@ -1,9 +1,10 @@
services:
docling_app: docling_app:
build: build:
context: . context: .
args: args:
ENABLE_CUDA: '1' ENABLE_CUDA: '1'
container_name: gradio_app container_name: docling_app
restart: unless-stopped restart: unless-stopped
ports: ports:
- "7880:7860" # Gradio application port - "7880:7860" # Gradio application port
@ -11,13 +12,13 @@
- ./app/services/weaviate_service.py:/gradio_app/weaviate_service.py - ./app/services/weaviate_service.py:/gradio_app/weaviate_service.py
environment: environment:
ENABLE_CUDA: '1' ENABLE_CUDA: '1'
NVIDIA_VISIBLE_DEVICES: '3' NVIDIA_VISIBLE_DEVICES: '2'
deploy: deploy:
resources: resources:
reservations: reservations:
devices: devices:
- driver: nvidia - driver: nvidia
device_ids: ['3'] # Specify GPU 3 device_ids: ['2'] # Specify GPU 3
capabilities: [gpu] capabilities: [gpu]
shm_size: 1g shm_size: 1g
depends_on: depends_on: