bmas10 commited on
Commit
921b9a7
·
verified ·
1 Parent(s): 03a30d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import torch
9
 
10
 
11
  model_name = "meta-llama/Llama-2-7b-chat-hf"
12
- access_token = base64.b64decode('aGZfcGlSVlFEZ0lTSUFOUm1abnVtQXZyTVlET1FnR0R3VVpqaQ==') # Replace with your token
13
 
14
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=access_token)
15
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto", use_auth_token=access_token)
 
9
 
10
 
11
  model_name = "meta-llama/Llama-2-7b-chat-hf"
12
+ access_token = base64.b64decode('aGZfekJxa0pXZEh1bm90UEJXek1mdkdOc096WXdIVVZvYkRwcg==') # Replace with your token
13
 
14
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=access_token)
15
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto", use_auth_token=access_token)