Spaces:
Runtime error
Runtime error
lambdabrendan
commited on
Commit
•
eef78f7
1
Parent(s):
ed88147
Remove alternate GPU vendors from drop-down - leaving logic incase we ever offer AMD
Browse files- src/app.py +3 -3
src/app.py
CHANGED
@@ -7,7 +7,7 @@ import numpy as np
|
|
7 |
import gc
|
8 |
from huggingface_hub import login
|
9 |
|
10 |
-
st.set_page_config(page_title='Lambda LLM Calculator', layout="wide", initial_sidebar_state="expanded")
|
11 |
|
12 |
model_list = [
|
13 |
"mistralai/Mistral-7B-v0.1",
|
@@ -36,7 +36,7 @@ model_list = [
|
|
36 |
"tiiuae/falcon-180B",
|
37 |
"tiiuae/falcon-180B-Chat",
|
38 |
]
|
39 |
-
st.title("
|
40 |
|
41 |
percentage_width_main = 80
|
42 |
st.markdown(
|
@@ -145,7 +145,7 @@ if model_name not in st.session_state:
|
|
145 |
st.session_state['actual_model'] = model_name
|
146 |
|
147 |
|
148 |
-
gpu_vendor = st.sidebar.selectbox("GPU Vendor", ["NVIDIA"
|
149 |
# year = st.sidebar.selectbox("Filter by Release Year", list(range(2014, 2024))[::-1], index=None)
|
150 |
gpu_info = gpu_specs[gpu_specs['Vendor'] == gpu_vendor].sort_values('Product Name')
|
151 |
# if year:
|
|
|
7 |
import gc
|
8 |
from huggingface_hub import login
|
9 |
|
10 |
+
st.set_page_config(page_title='Lambda LLM VRAM Calculator', layout="wide", initial_sidebar_state="expanded")
|
11 |
|
12 |
model_list = [
|
13 |
"mistralai/Mistral-7B-v0.1",
|
|
|
36 |
"tiiuae/falcon-180B",
|
37 |
"tiiuae/falcon-180B-Chat",
|
38 |
]
|
39 |
+
st.title("Lambda LLM VRAM Calculator")
|
40 |
|
41 |
percentage_width_main = 80
|
42 |
st.markdown(
|
|
|
145 |
st.session_state['actual_model'] = model_name
|
146 |
|
147 |
|
148 |
+
gpu_vendor = st.sidebar.selectbox("GPU Vendor", ["NVIDIA"])
|
149 |
# year = st.sidebar.selectbox("Filter by Release Year", list(range(2014, 2024))[::-1], index=None)
|
150 |
gpu_info = gpu_specs[gpu_specs['Vendor'] == gpu_vendor].sort_values('Product Name')
|
151 |
# if year:
|