add package ollama

modelscope · Mar 12, 2024 · 8a0678f · 8a0678f
1 parent 0d30a14
commit 8a0678f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -44,12 +44,14 @@
     "tiktoken",
     "Pillow",
     "requests",
-    "openai>=1.3.0",
     "numpy",
     "Flask==3.0.0",
     "Flask-Cors==4.0.0",
     "Flask-SocketIO==5.3.6",
+    # TODO: move into other requires
     "dashscope==1.14.1",
+    "openai>=1.3.0",
+    "ollama>=0.1.7"
 ]
 
 distribute_requires = minimal_requires + rpc_requires