rjiang12 commited on
Commit
cbb47a2
·
1 Parent(s): 2fb15b1

Update openAI_manager.py

Browse files
Files changed (1) hide show
  1. openAI_manager.py +2 -1
openAI_manager.py CHANGED
@@ -83,9 +83,10 @@ def generateTestSentences(model_name, group_kwds, attribute_kwds, progress):
83
  kwd_pair = [grp_kwd.strip(), att_kwd.strip()]
84
 
85
  gens = genChatGPT(model_name, kwd_pair, num2gen, numTries, fixed_shots, temperature=0.8)
86
- print(f"Gens for pair: <{kwd_pair}> -> {gens}")
87
  # progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"{gens}")
88
  for gen in gens:
 
89
  progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"{gen[2]}")
90
  all_gens.extend(gens)
91
 
 
83
  kwd_pair = [grp_kwd.strip(), att_kwd.strip()]
84
 
85
  gens = genChatGPT(model_name, kwd_pair, num2gen, numTries, fixed_shots, temperature=0.8)
86
+ # print(f"Gens for pair: <{kwd_pair}> -> {gens}")
87
  # progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"{gens}")
88
  for gen in gens:
89
+ print(gen)
90
  progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"{gen[2]}")
91
  all_gens.extend(gens)
92