Spaces:
Sleeping
Sleeping
Update llm/utils.py
Browse files- llm/utils.py +1 -3
llm/utils.py
CHANGED
|
@@ -49,9 +49,7 @@ def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vdb,
|
|
| 49 |
|
| 50 |
# Initialize LLM
|
| 51 |
def initialize_LLM(llm_temperature, max_tokens, top_k, vector_db, thold, progress=gr.Progress()):
|
| 52 |
-
|
| 53 |
-
llm_name = "mistralai/Mistral-7B-Instruct-v0.2"
|
| 54 |
-
#print("llm_name: ",llm_name)
|
| 55 |
qa_chain = initialize_llmchain(llm_name, llm_temperature, max_tokens, top_k, vector_db, thold)
|
| 56 |
return qa_chain
|
| 57 |
|
|
|
|
| 49 |
|
| 50 |
# Initialize LLM
|
| 51 |
def initialize_LLM(llm_temperature, max_tokens, top_k, vector_db, thold, progress=gr.Progress()):
|
| 52 |
+
llm_name = "meta-llama/Meta-Llama-3-8B-Instruct" #"mistralai/Mistral-7B-Instruct-v0.2"
|
|
|
|
|
|
|
| 53 |
qa_chain = initialize_llmchain(llm_name, llm_temperature, max_tokens, top_k, vector_db, thold)
|
| 54 |
return qa_chain
|
| 55 |
|