added a list of some models

requirements.txt

@@ -3,4 +3,5 @@ transformers @ git+https://github.com/huggingface/transformers
 huggingface_hub
 pandas
 plotly
+streamlit-datalist
 einops==0.6.1

src/app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+from streamlit_datalist import stDatalist
 import pandas as pd
 from utils import extract_from_url, get_model, calculate_memory
 import plotly.express as px
@@ -7,6 +8,33 @@ import gc
 
 st.set_page_config(page_title='Can you run it? LLM version', layout="wide", initial_sidebar_state="expanded")
 
+model_list = [
+    "mistralai/Mistral-7B-v0.1",
+    "mistralai/Mistral-7B-Instruct-v0.1",
+    "ehartford/samantha-mistral-7b",
+    "SkunkworksAI/Mistralic-7B-1",
+    "microsoft/phi-1_5",
+    "PY007/TinyLlama-1.1B-intermediate-step-480k-1T"
+    "codellama/CodeLlama-7b-hf",
+    "codellama/CodeLlama-13b-hf",
+    "codellama/CodeLlama-34b-hf",
+    "Phind/Phind-CodeLlama-34B-v2",
+    "WizardLM/WizardCoder-Python-34B-V1.0",
+    "TheBloke/Llama-2-7B-fp16",
+    "TheBloke/Llama-2-13B-fp16",
+    "TheBloke/Llama-2-70B-fp16",
+    "Gryphe/MythoMax-L2-13b",
+    "uukuguy/speechless-llama2-hermes-orca-platypus-wizardlm-13b",
+    "lmsys/vicuna-7b-v1.5",
+    "lmsys/vicuna-13b-v1.5-16k",
+    "lmsys/longchat-7b-v1.5-32k",
+    "tiiuae/falcon-7B-Instruct",
+    "tiiuae/falcon-7B",
+    "tiiuae/falcon-40B",
+    "tiiuae/falcon-40B-Instruct",
+    "tiiuae/falcon-180B",
+    "tiiuae/falcon-180B-Chat",
+]
 st.title("Can you run it? LLM version")
 
 percentage_width_main = 80
@@ -77,7 +105,9 @@ with col.expander("Information", expanded=True):
     st.latex(r"\text{Memory}_\text{LoRa} \approx \text{Model Size} + \left(\text{ \# trainable Params}_\text{Billions}\times\frac{16}{8} \times 4\right) \times 1.2")
 
 access_token = st.sidebar.text_input("Access token")
-model_name = st.sidebar.text_input("Model name", value="mistralai/Mistral-7B-v0.1")
+#model_name = st.sidebar.text_input("Model name", value="mistralai/Mistral-7B-v0.1")
+with st.sidebar.container():
+    model_name  = stDatalist("Model name (Press Enter to apply)", model_list, index=0)
 if not model_name:
     st.info("Please enter a model name")
     st.stop()