ag-mach commited on
Commit
caf45a8
·
1 Parent(s): 45d41ea

switch to smaller model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def init():
23
 
24
  def init_llm_pipeline():
25
  if "llm" not in st.session_state:
26
- model_id = "bigcode/starcoder2-15b"
27
 
28
  tokenizer = AutoTokenizer.from_pretrained(model_id)
29
  model = AutoModelForCausalLM.from_pretrained(
 
23
 
24
  def init_llm_pipeline():
25
  if "llm" not in st.session_state:
26
+ model_id = "bigcode/starcoder2-7b"
27
 
28
  tokenizer = AutoTokenizer.from_pretrained(model_id)
29
  model = AutoModelForCausalLM.from_pretrained(