feat[bert]: FastAPI interface
Browse files- docker-compose.yml +3 -1
docker-compose.yml
CHANGED
|
@@ -1,12 +1,14 @@
|
|
| 1 |
services:
|
| 2 |
ml-service:
|
| 3 |
build: ./ml-service
|
|
|
|
| 4 |
ports:
|
| 5 |
- 7860:7860
|
| 6 |
deploy:
|
| 7 |
resources:
|
| 8 |
limits:
|
| 9 |
cpus: '0.80' # Limita a 80% de la CPU
|
| 10 |
-
memory: 6G # Limita a
|
|
|
|
| 11 |
devices:
|
| 12 |
- "/dev/nvidia0:/dev/nvidia0"
|
|
|
|
| 1 |
services:
|
| 2 |
ml-service:
|
| 3 |
build: ./ml-service
|
| 4 |
+
shm_size: '2gb' # Limita a 2 GB de memoria compartida
|
| 5 |
ports:
|
| 6 |
- 7860:7860
|
| 7 |
deploy:
|
| 8 |
resources:
|
| 9 |
limits:
|
| 10 |
cpus: '0.80' # Limita a 80% de la CPU
|
| 11 |
+
memory: 6G # Limita a 6 GB de RAM
|
| 12 |
+
|
| 13 |
devices:
|
| 14 |
- "/dev/nvidia0:/dev/nvidia0"
|