Update networks/message_streamer.py

networks/message_streamer.py

@@ -21,6 +21,7 @@ class MessageStreamer:
         "mistral-7b": "</s>",
         "nous-mixtral-8x7b": "<|im_end|>",
         "openchat-3.5": "<|end_of_turn|>",
+        "codellama-7b": "<|end_of_turn|>",
         "gemma-7b": "<eos>",
     }
     TOKEN_LIMIT_MAP = {
@@ -28,6 +29,7 @@ class MessageStreamer:
         "mistral-7b": 32768,
         "nous-mixtral-8x7b": 32768,
         "openchat-3.5": 8192,
+        "codellama-7b": 8192,
         "gemma-7b": 8192,
     }
     TOKEN_RESERVED = 100