Spaces:

ka1kuk
/

LLM-api

Sleeping

ka1kuk commited on Jan 14, 2024

Commit

b7a84dd

verified ·

1 Parent(s): 6cede14

Update networks/message_streamer.py

Files changed (1) hide show

networks/message_streamer.py CHANGED Viewed

@@ -26,9 +26,9 @@ class MessageStreamer:
             self.model = model
         else:
             self.model = "default"
-        self.model_fullname = self.MODEL_MAP[self.model]
-        self.message_outputer = OpenaiStreamOutputer()
-        self.stop_sequences = self.STOP_SEQUENCES_MAP.get(self.model, "")
     def parse_line(self, line):
         try:

             self.model = model
         else:
             self.model = "default"
+            self.model_fullname = self.MODEL_MAP[self.model]
+            self.message_outputer = OpenaiStreamOutputer()
+            self.stop_sequences = self.STOP_SEQUENCES_MAP.get(self.model, "")
     def parse_line(self, line):
         try: