sandro-timia commited on
Commit
c11b06b
1 Parent(s): 727eabd
Files changed (1) hide show
  1. utils/llama_index_utils.py +1 -1
utils/llama_index_utils.py CHANGED
@@ -13,7 +13,7 @@ def initialize_llama_settings():
13
  model_name="google/gemma-1.1-7b-it",
14
  tokenizer_name="google/gemma-1.1-7b-it",
15
  context_window=3000,
16
- token=os.getenv("HF_TOKEN"),
17
  max_new_tokens=512,
18
  generate_kwargs={"temperature": 0.1},
19
  )
 
13
  model_name="google/gemma-1.1-7b-it",
14
  tokenizer_name="google/gemma-1.1-7b-it",
15
  context_window=3000,
16
+ token=os.getenv("timia-framework"),
17
  max_new_tokens=512,
18
  generate_kwargs={"temperature": 0.1},
19
  )