Daniel Marques commited on
Commit
b2d865e
·
1 Parent(s): 1d1dd8d

fix: add console trupple

Browse files
Files changed (1) hide show
  1. main.py +3 -0
main.py CHANGED
@@ -45,6 +45,9 @@ DB = Chroma(
45
  RETRIEVER = DB.as_retriever()
46
 
47
  models = load_model(device_type=DEVICE_TYPE, model_id=MODEL_ID, model_basename=MODEL_BASENAME, stream=False)
 
 
 
48
  LLM, STREAMER = models
49
 
50
  template = """Your name is Katara and you are a helpful, respectful and honest assistant. You should only use the source documents provided to answer the questions.
 
45
  RETRIEVER = DB.as_retriever()
46
 
47
  models = load_model(device_type=DEVICE_TYPE, model_id=MODEL_ID, model_basename=MODEL_BASENAME, stream=False)
48
+
49
+ print(models)
50
+
51
  LLM, STREAMER = models
52
 
53
  template = """Your name is Katara and you are a helpful, respectful and honest assistant. You should only use the source documents provided to answer the questions.