zhengr commited on
Commit
3457891
·
verified ·
1 Parent(s): 9869e27

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
  # load model
21
  #model = "NousResearch/Llama-2-7b-hf"
22
- model = "unsloth/llama-2-7b-chat-bnb-4bit"
23
  peft_model = "FinGPT/fingpt-forecaster_sz50_llama2-7B_lora"
24
 
25
  tokenizer = AutoTokenizer.from_pretrained(model, token = access_token, device_map="auto", trust_remote_code=True)
 
19
 
20
  # load model
21
  #model = "NousResearch/Llama-2-7b-hf"
22
+ model = "meta-llama/Llama-2-7b-chat-hf"
23
  peft_model = "FinGPT/fingpt-forecaster_sz50_llama2-7B_lora"
24
 
25
  tokenizer = AutoTokenizer.from_pretrained(model, token = access_token, device_map="auto", trust_remote_code=True)