zhtet commited on
Commit
4335fe0
·
1 Parent(s): 34ae42f

Update models/llamaCustom.py

Browse files
Files changed (1) hide show
  1. models/llamaCustom.py +4 -4
models/llamaCustom.py CHANGED
@@ -183,10 +183,10 @@ class LlamaCustom:
183
 
184
  def get_response(self, query_str):
185
  print("query_str: ", query_str)
186
- # query_engine = self.vector_index.as_query_engine()
187
- query_engine = self.vector_index.as_query_engine(
188
- text_qa_template=text_qa_template, refine_template=refine_template
189
- )
190
  response = query_engine.query(query_str)
191
  print("metadata: ", response.metadata)
192
  return str(response)
 
183
 
184
  def get_response(self, query_str):
185
  print("query_str: ", query_str)
186
+ query_engine = self.vector_index.as_query_engine()
187
+ # query_engine = self.vector_index.as_query_engine(
188
+ # text_qa_template=text_qa_template, refine_template=refine_template
189
+ # )
190
  response = query_engine.query(query_str)
191
  print("metadata: ", response.metadata)
192
  return str(response)