Switch to onnxruntime-gpu

laurynas · Dec 14, 2024 · f524486 · f524486
1 parent 3732fc6
commit f524486
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-onnxruntime==1.17.0
+onnxruntime-gpu==1.20.1
 pillow
 Flask==3.0.2
 gunicorn==21.2.0