Saif Rehman Nasir
commited on
Commit
·
a26ec13
1
Parent(s):
ac80634
Refactor code
Browse files
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 |
-
|
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###
|