Spaces:
Paused
Paused
Commit
·
090e312
1
Parent(s):
6632240
Update llm.py
Browse files
llm.py
CHANGED
@@ -28,10 +28,7 @@ from utils import *
|
|
28 |
import logging
|
29 |
import argparse
|
30 |
|
31 |
-
|
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'):
|