zhengr commited on
Commit
cc997c6
·
verified ·
1 Parent(s): cf05f7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ access_token = os.environ["TOKEN"]
19
 
20
  base_model = "NousResearch/Llama-2-13b-hf"
21
  peft_model = "FinGPT/fingpt-sentiment_llama2-13b_lora"
22
- tokenizer = LlamaTokenizerFast.from_pretrained(base_model, trust_remote_code=True)
23
  tokenizer.pad_token = tokenizer.eos_token
24
  model = LlamaForCausalLM.from_pretrained(base_model, trust_remote_code=True, device_map = "cuda:0", load_in_8bit = True,)
25
  model = PeftModel.from_pretrained(model, peft_model)
 
19
 
20
  base_model = "NousResearch/Llama-2-13b-hf"
21
  peft_model = "FinGPT/fingpt-sentiment_llama2-13b_lora"
22
+ tokenizer = AutoTokenizer.from_pretrained(base_model, trust_remote_code=True)
23
  tokenizer.pad_token = tokenizer.eos_token
24
  model = LlamaForCausalLM.from_pretrained(base_model, trust_remote_code=True, device_map = "cuda:0", load_in_8bit = True,)
25
  model = PeftModel.from_pretrained(model, peft_model)