From ece2d880aa11a20aec62c53bb0c94c236e7eeb18 Mon Sep 17 00:00:00 2001
From: Boon <49676590+bkiat1123@users.noreply.github.com>
Date: Wed, 20 Mar 2024 15:20:44 +0800
Subject: [PATCH] Add ollama to setup.py

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index 1cb48f4..2a282f4 100644
--- a/setup.py
+++ b/setup.py
@@ -31,6 +31,7 @@
         "grpcio~=1.54.2",
         "google-generativeai",
         "mistralai",
+        "ollama",
     ],
     extras_require={
         "local": ["einops", "accelerate"]