Spaces:

Pro100Sata
/

xche_ai

Sleeping

Yarik commited on May 26

Commit

dc1cc98

•

1 Parent(s): 9125999

Update space

Files changed (1) hide show

networks/message_streamer.py CHANGED Viewed

@@ -25,7 +25,7 @@ class MessageStreamer:
         "mistral-7b": "</s>",
         "nous-mixtral-8x7b": "<|im_end|>",
         "zephyr-7b-beta": "</s>",
-        "starchat2-15b-v0.1": "</s>",
     }
     TOKEN_LIMIT_MAP = {
         "mixtral-8x7b": 32768,

         "mistral-7b": "</s>",
         "nous-mixtral-8x7b": "<|im_end|>",
         "zephyr-7b-beta": "</s>",
+        "starchat2-15b-v0.1": "<|im_end|>",
     }
     TOKEN_LIMIT_MAP = {
         "mixtral-8x7b": 32768,