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

added EvalDataset Generation

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -20,6 +20,7 @@ import threading
20
  import re
21
  import csv
22
  import json
 
23
 
24
  from openai import OpenAI
25
  from langchain.text_splitter import RecursiveCharacterTextSplitter
@@ -718,6 +719,8 @@ class BSIChatbot:
718
  for data in data_list:
719
  print("Eval Entry no:")
720
  print(i)
 
 
721
  print("frage:")
722
  print(data["Frage"])
723
  #def ragPromptNew(self, query, rerankingStep, history, stepBackPrompt)
 
20
  import re
21
  import csv
22
  import json
23
+ import gc
24
 
25
  from openai import OpenAI
26
  from langchain.text_splitter import RecursiveCharacterTextSplitter
 
719
  for data in data_list:
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)