From c350f8049db495dbb5305d3ffe8a9ece698ddfdb Mon Sep 17 00:00:00 2001 From: Mambo56 <91324873+Mambo56@users.noreply.github.com> Date: Tue, 9 Apr 2024 16:36:02 +0300 Subject: [PATCH] Update kab.py --- streamlit/kab.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/streamlit/kab.py b/streamlit/kab.py index 1d8e952..32fe830 100644 --- a/streamlit/kab.py +++ b/streamlit/kab.py @@ -30,7 +30,7 @@ def icon(emoji: str): models = { "llama2-70b-4096": {"name": "LLaMA2-70b-chat", "tokens": 4096, "developer": "Meta"}, "mixtral-8x7b-32768": {"name": "Mixtral-8x7b-Instruct-v0.1", "tokens": 32768, "developer": "Mistral"} - "gemma-7b-it": {"name": "Gemma-7b-it", "tokens": 8192, "developer": "Google"} + } # Layout for model selection and max_tokens slider