Spaces:
Sleeping
Sleeping
XufengDuan
commited on
Commit
•
0e76736
1
Parent(s):
98c5aed
update scripts
Browse files
src/backend/model_operations.py
CHANGED
@@ -178,7 +178,7 @@ class SummaryGenerator:
|
|
178 |
if 'Stimuli-2' in df_sheet.columns:
|
179 |
Stimuli_2_column = df_sheet["Stimuli-2"]
|
180 |
|
181 |
-
for j, prompt_value in enumerate(tqdm(prompt_column
|
182 |
if exit_outer_loop:
|
183 |
break
|
184 |
ID = 'E' + str(i)
|
@@ -187,7 +187,7 @@ class SummaryGenerator:
|
|
187 |
# print(ID, q_ID, prompt_value)
|
188 |
system_prompt = envs.SYSTEM_PROMPT
|
189 |
_user_prompt = prompt_value
|
190 |
-
for ii in range(
|
191 |
# user_prompt = f"{envs.USER_PROMPT}\nPassage:\n{_source}"
|
192 |
while True:
|
193 |
try:
|
@@ -1253,7 +1253,7 @@ class EvaluationModel:
|
|
1253 |
|
1254 |
envs.API.upload_file(
|
1255 |
path_or_fileobj= save_path,#./generation_results/meta-llama/Llama-2-13b-chat-hf_coding.csv
|
1256 |
-
path_in_repo=f"{save_path.replace('generation_results/','')}
|
1257 |
repo_id=envs.RESULTS_REPO,
|
1258 |
repo_type="dataset",
|
1259 |
)
|
|
|
178 |
if 'Stimuli-2' in df_sheet.columns:
|
179 |
Stimuli_2_column = df_sheet["Stimuli-2"]
|
180 |
|
181 |
+
for j, prompt_value in enumerate(tqdm(prompt_column, desc=f"Processing {sheet_name}"), start=0):
|
182 |
if exit_outer_loop:
|
183 |
break
|
184 |
ID = 'E' + str(i)
|
|
|
187 |
# print(ID, q_ID, prompt_value)
|
188 |
system_prompt = envs.SYSTEM_PROMPT
|
189 |
_user_prompt = prompt_value
|
190 |
+
for ii in range(10):
|
191 |
# user_prompt = f"{envs.USER_PROMPT}\nPassage:\n{_source}"
|
192 |
while True:
|
193 |
try:
|
|
|
1253 |
|
1254 |
envs.API.upload_file(
|
1255 |
path_or_fileobj= save_path,#./generation_results/meta-llama/Llama-2-13b-chat-hf_coding.csv
|
1256 |
+
path_in_repo=f"{save_path.replace('generation_results/','')}",#
|
1257 |
repo_id=envs.RESULTS_REPO,
|
1258 |
repo_type="dataset",
|
1259 |
)
|