VLog4CustomLLMsPlusQA-3

Sleeping

dj86 commited on Sep 3

Commit

544aa00

•

1 Parent(s): 4fe26d6

Update vlog4chat.py

Files changed (1) hide show

vlog4chat.py CHANGED Viewed

@@ -129,7 +129,7 @@ class Vlogger4chat :
     def init_llm(self):
         print('\033[1;33m' + "Initializing LLM Reasoner...".center(50, '-') + '\033[0m')
         self.llm = TogetherLLM(
-            model= os.environ("YOUR_MODEL_NAME"),
             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
         )