matthoffner commited on
Commit
090e312
·
1 Parent(s): 6632240

Update llm.py

Browse files
Files changed (1) hide show
  1. llm.py +1 -4
llm.py CHANGED
@@ -28,10 +28,7 @@ from utils import *
28
  import logging
29
  import argparse
30
 
31
- parser = argparse.ArgumentParser()
32
- parser.add_argument('--model', type=str, required=True, default="wizardLM-7B.ggml.q5_0.bin")
33
- args = parser.parse_args()
34
- model_path = args.model
35
 
36
 
37
  def query_llm(index, prompt, service_context, retriever_mode='embedding', response_mode='tree_summarize'):
 
28
  import logging
29
  import argparse
30
 
31
+ model_path = "wizardLM-7B.ggml.q5_0.bin"
 
 
 
32
 
33
 
34
  def query_llm(index, prompt, service_context, retriever_mode='embedding', response_mode='tree_summarize'):