Gainward777 commited on
Commit
69bdb2e
·
verified ·
1 Parent(s): 9227aa1

Update llm/utils.py

Browse files
Files changed (1) hide show
  1. 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
- # print("llm_option",llm_option)
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