MikeMann commited on
Commit
81b71b0
·
1 Parent(s): db6fe0f

added EvalDataset Generation

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -283,7 +283,7 @@ class BSIChatbot:
283
  docstore = vectorstore.docstore._dict.values()
284
 
285
  #for doc_id in vectorstore.index_to_docstore_id.values():
286
- for entry in docs:
287
  # Hole das Dokument aus dem docstore
288
  #print("DBG:"+doc_id)
289
  #document = vectorstore.get_by_ids(doc_id)
@@ -720,7 +720,7 @@ class BSIChatbot:
720
  print("Eval Entry no:")
721
  print(i)
722
  print("GPU Memory Allocated:")
723
- print(torch.cuda.memory_allocated())
724
  print("frage:")
725
  print(data["Frage"])
726
  #def ragPromptNew(self, query, rerankingStep, history, stepBackPrompt)
 
283
  docstore = vectorstore.docstore._dict.values()
284
 
285
  #for doc_id in vectorstore.index_to_docstore_id.values():
286
+ for entry in docstore:
287
  # Hole das Dokument aus dem docstore
288
  #print("DBG:"+doc_id)
289
  #document = vectorstore.get_by_ids(doc_id)
 
720
  print("Eval Entry no:")
721
  print(i)
722
  print("GPU Memory Allocated:")
723
+ print(torch.cuda.memory_allocated()/1024/1024/1024)
724
  print("frage:")
725
  print(data["Frage"])
726
  #def ragPromptNew(self, query, rerankingStep, history, stepBackPrompt)