Spaces:
Running
Running
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 |
|