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
Files changed (1) hide show
  1. 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("Can you run it? LLM version")
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", "AMD", "Intel", "Apple"])
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: