fix compose
This commit is contained in:
parent
671f475e11
commit
be8aaff73c
|
|
@ -1,9 +1,10 @@
|
|||
services:
|
||||
docling_app:
|
||||
build:
|
||||
context: .
|
||||
args:
|
||||
ENABLE_CUDA: '1'
|
||||
container_name: gradio_app
|
||||
container_name: docling_app
|
||||
restart: unless-stopped
|
||||
ports:
|
||||
- "7880:7860" # Gradio application port
|
||||
|
|
@ -11,13 +12,13 @@
|
|||
- ./app/services/weaviate_service.py:/gradio_app/weaviate_service.py
|
||||
environment:
|
||||
ENABLE_CUDA: '1'
|
||||
NVIDIA_VISIBLE_DEVICES: '3'
|
||||
NVIDIA_VISIBLE_DEVICES: '2'
|
||||
deploy:
|
||||
resources:
|
||||
reservations:
|
||||
devices:
|
||||
- driver: nvidia
|
||||
device_ids: ['3'] # Specify GPU 3
|
||||
device_ids: ['2'] # Specify GPU 3
|
||||
capabilities: [gpu]
|
||||
shm_size: 1g
|
||||
depends_on:
|
||||
|
|
|
|||
Loading…
Reference in New Issue