Add hybrid deps (#253)

onnx · Dec 21, 2024 · 5432f72 · 5432f72
1 parent e0ba4b7
commit 5432f72
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -100,11 +100,14 @@
             "fastapi",
             "uvicorn[standard]",
         ],
-	"llm-oga-hybrid": [
+        "llm-oga-hybrid": [
             "transformers",
             "torch",
             "onnx==1.16.1",
             "numpy==1.26.4",
+            "datasets",
+            "fastapi",
+            "uvicorn[standard]",
         ],
     },
     classifiers=[],