Saif Rehman Nasir commited on
Commit
a26ec13
·
1 Parent(s): ac80634

Refactor code

Browse files
Files changed (1) hide show
  1. rag.py +3 -3
rag.py CHANGED
@@ -28,6 +28,8 @@ chat_llm = HuggingFaceEndpoint(
28
  do_sample=False,
29
  )
30
 
 
 
31
 
32
  def local_retriever(query: str):
33
  topChunks = 3
@@ -274,9 +276,7 @@ def global_retriever(query: str, level: int, response_type: str):
274
  i += 1
275
 
276
  ###Debug####
277
- global_tokenizer = LlamaTokenizer.from_pretrained(
278
- "meta-llama/Meta-Llama-3-8B-Instruct"
279
- )
280
  tokens = global_tokenizer.encode(intermediate_results)
281
  print(f"Number of input tokens: {len(tokens)}")
282
  ###Debug###
 
28
  do_sample=False,
29
  )
30
 
31
+ global_tokenizer = LlamaTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
32
+
33
 
34
  def local_retriever(query: str):
35
  topChunks = 3
 
276
  i += 1
277
 
278
  ###Debug####
279
+
 
 
280
  tokens = global_tokenizer.encode(intermediate_results)
281
  print(f"Number of input tokens: {len(tokens)}")
282
  ###Debug###