Spaces:
Sleeping
Sleeping
Update openAI_manager.py
Browse files- openAI_manager.py +3 -1
openAI_manager.py
CHANGED
@@ -84,7 +84,9 @@ def generateTestSentences(model_name, group_kwds, attribute_kwds, progress):
|
|
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 |
all_gens.extend(gens)
|
89 |
|
90 |
return all_gens
|
|
|
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}")
|
90 |
all_gens.extend(gens)
|
91 |
|
92 |
return all_gens
|