From 0e7468814517ebdccac568a3547b9d0353a02acb Mon Sep 17 00:00:00 2001 From: Rohit Gupta Date: Thu, 6 Jul 2023 01:01:32 +0530 Subject: [PATCH] update: import gpt4all locally (#146) Import gpt4all locally to avoid error in case its not present. --- embedchain/embedchain.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/embedchain/embedchain.py b/embedchain/embedchain.py index f13b507d..e4a49c5a 100644 --- a/embedchain/embedchain.py +++ b/embedchain/embedchain.py @@ -3,7 +3,6 @@ import os from chromadb.utils import embedding_functions from dotenv import load_dotenv -from gpt4all import GPT4All from langchain.docstore.document import Document from langchain.embeddings.openai import OpenAIEmbeddings @@ -292,6 +291,8 @@ class OpenSourceApp(EmbedChain): super().__init__(db, ef) def get_llm_model_answer(self, prompt): + from gpt4all import GPT4All + global gpt4all_model if gpt4all_model is None: gpt4all_model = GPT4All("orca-mini-3b.ggmlv3.q4_0.bin")