XufengDuan commited on
Commit
0e76736
1 Parent(s): 98c5aed

update scripts

Browse files
Files changed (1) hide show
  1. src/backend/model_operations.py +3 -3
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[0:2], desc=f"Processing {sheet_name}"), start=0):
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(1):
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/','')}_coding.csv",#
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
  )