sravan commited on
Commit
bd9af43
·
1 Parent(s): 78bb570
Files changed (1) hide show
  1. chains.py +2 -2
chains.py CHANGED
@@ -17,10 +17,10 @@ from transformers import AutoTokenizer
17
 
18
  data_indexer = DataIndexer()
19
 
20
- tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
21
 
22
  llm = HuggingFaceEndpoint(
23
- repo_id="meta-llama/Meta-Llama-3-8B-Instruct",
24
 
25
  huggingfacehub_api_token=os.environ['HF_TOKEN'],
26
  max_new_tokens=512,
 
17
 
18
  data_indexer = DataIndexer()
19
 
20
+ tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3.1-8B-Instruct")
21
 
22
  llm = HuggingFaceEndpoint(
23
+ repo_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
24
 
25
  huggingfacehub_api_token=os.environ['HF_TOKEN'],
26
  max_new_tokens=512,