refactor: default to Gemma model

datasciencecampus · Aug 20, 2024 · 70a5e25 · 70a5e25
1 parent 0d69f64
commit 70a5e25
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/parliai_public/readers/base.py b/src/parliai_public/readers/base.py
@@ -342,6 +342,8 @@ def _read_contents(self, soup: BeautifulSoup) -> dict:
     def instantiate_llm(self) -> None:
         """Instantiate LLM object per user specification."""
 
+        # Temporary override to default to Gemma (known/tested LLM)
+        self.llm_name = "gemma"
         self.llm = ChatOllama(model=self.llm_name, temperature=0)
 
         return None