keep only working llm models

config.py

@@ -1,9 +1,11 @@
 MODELS={
     "DEFAULT":"tiiuae/falcon-7b-instruct",
-    "gpt4all":"gpt4all",
+    # "gpt4all":"gpt4all",
     "flan-t5-xxl":"google/flan-t5-xxl",
     "falcon-7b-instruct":"tiiuae/falcon-7b-instruct",
     "openai gpt-3.5":"openai",
+    # "Deci/DeciLM-6b-instruct":"Deci/DeciLM-6b-instruct",
+    # "Deci/DeciLM-6b":"Deci/DeciLM-6b",
 
 }
 

qaPipeline.py

@@ -148,6 +148,10 @@ class QAPipeline:
                     self.llm = HuggingFaceHub(repo_id=model_type, model_kwargs={"temperature":0.001, "max_length":1024})
                 case "openai":
                     self.llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0)
+                case "Deci/DeciLM-6b-instruct":
+                    self.llm = ChatOpenAI(model_name="Deci/DeciLM-6b-instruct", temperature=0)
+                case "Deci/DeciLM-6b":
+                    self.llm = ChatOpenAI(model_name="Deci/DeciLM-6b", temperature=0)
                 case _default:
                     # raise exception if model_type is not supported
                     raise Exception(f"Model type {model_type} is not supported. Please choose a valid one")