13thapostle commited on
Commit
efdf91b
·
verified ·
1 Parent(s): d71c5a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,7 +19,7 @@ access_token = os.environ["HF_TOKEN"]
19
  finnhub_client = finnhub.Client(api_key=os.environ["FINNHUB_API_KEY"])
20
 
21
  base_model = AutoModelForCausalLM.from_pretrained(
22
- 'meta-llama/Llama-2-7b',
23
  token=access_token,
24
  trust_remote_code=True,
25
  device_map="auto",
@@ -34,7 +34,7 @@ model = PeftModel.from_pretrained(
34
  model = model.eval()
35
 
36
  tokenizer = AutoTokenizer.from_pretrained(
37
- 'meta-llama/Llama-2-7b',
38
  token=access_token
39
  )
40
 
 
19
  finnhub_client = finnhub.Client(api_key=os.environ["FINNHUB_API_KEY"])
20
 
21
  base_model = AutoModelForCausalLM.from_pretrained(
22
+ 'meta-llama/Llama-2-7b-chat-hf',
23
  token=access_token,
24
  trust_remote_code=True,
25
  device_map="auto",
 
34
  model = model.eval()
35
 
36
  tokenizer = AutoTokenizer.from_pretrained(
37
+ 'meta-llama/Llama-2-7b-chat-hf',
38
  token=access_token
39
  )
40