molba2see commited on
Commit
156b204
·
verified ·
1 Parent(s): 7b2c554

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -61,7 +61,7 @@ def load_my_model():
61
  tokenizer = LlamaTokenizerFast.from_pretrained(BASE_MODEL_NAME, trust_remote_code=True)
62
  tokenizer.pad_token = tokenizer.eos_token
63
 
64
- peft_model = PeftModel.from_pretrained(base_model, ADAPTER_PATH, is_trainable=True)
65
  return peft_model, tokenizer
66
 
67
 
 
61
  tokenizer = LlamaTokenizerFast.from_pretrained(BASE_MODEL_NAME, trust_remote_code=True)
62
  tokenizer.pad_token = tokenizer.eos_token
63
 
64
+ peft_model = PeftModel.from_pretrained(base_model, ADAPTER_PATH)
65
  return peft_model, tokenizer
66
 
67