From 943d56d478e970407e2cd5355121ee0285c71042 Mon Sep 17 00:00:00 2001
From: _Yusaki <puree.marimjie@gmail.com>
Date: Fri, 13 Mar 2026 18:09:48 +0700
Subject: [PATCH] Add GPU passthrough for BERT model acceleration

---
 docker-compose.yml | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 3db56d6..1a8090d 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -12,4 +12,11 @@ services:
       - "5001"
     restart: unless-stopped
     volumes:
-      - ./backend/uploads:/app/backend/uploads
\ No newline at end of file
+      - ./backend/uploads:/app/backend/uploads
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
\ No newline at end of file