CD17 commited on
Commit
b19c896
·
verified ·
1 Parent(s): 82cfa19

Update functions_llama.py

Browse files
Files changed (1) hide show
  1. functions_llama.py +2 -2
functions_llama.py CHANGED
@@ -8,9 +8,9 @@ import re
8
  from typing import Dict, List
9
  import os
10
 
11
- token = os.getenv(LLAMA_TOKEN)
12
  if token is None:
13
- raise ValueError(LLAMA_TOKEN is not set in the environment variables)
14
  login(token=token)
15
 
16
  # Load model and tokenizer
 
8
  from typing import Dict, List
9
  import os
10
 
11
+ token = os.getenv('LLAMA_TOKEN')
12
  if token is None:
13
+ raise ValueError("LLAMA_TOKEN is not set in the environment variables")
14
  login(token=token)
15
 
16
  # Load model and tokenizer