VLog4CustomLLMsPlusQA-3

Sleeping

dj86 commited on Sep 3

Commit

05a283f

•

1 Parent(s): b430493

Update vlog4chat.py

Files changed (1) hide show

vlog4chat.py CHANGED Viewed

@@ -131,7 +131,7 @@ class Vlogger4chat :
     def init_llm(self):
         print('\033[1;33m' + "Initializing LLM Reasoner...".center(50, '-') + '\033[0m')
         self.llm = TogetherLLM(
-            model= "microsoft/WizardLM-2-8x22B",
             temperature=0.1,
             max_tokens=768
         )

     def init_llm(self):
         print('\033[1;33m' + "Initializing LLM Reasoner...".center(50, '-') + '\033[0m')
         self.llm = TogetherLLM(
+            model= os.getenv('Your_Model_Name'),
             temperature=0.1,
             max_tokens=768
         )