Wendy commited on
Commit
0d409d1
·
verified ·
1 Parent(s): 945a97b

Upload cogagent_infer_batch.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. cogagent_infer_batch.py +4 -1
cogagent_infer_batch.py CHANGED
@@ -159,7 +159,10 @@ def main():
159
  history_str += f"\n{index}. {step}\t{action}"
160
 
161
  # Compose the query with task, platform, and selected format instructions
162
- query = f"Task: {task}{history_str}\n{platform_str}{format_str}"
 
 
 
163
 
164
  #print(f"Round {round_num} query:\n{query}")
165
 
 
159
  history_str += f"\n{index}. {step}\t{action}"
160
 
161
  # Compose the query with task, platform, and selected format instructions
162
+ query = []
163
+ for x in task:
164
+ query.append(f"Task: {x}{history_str}\n{platform_str}{format_str}")
165
+ #query = f"Task: {task}{history_str}\n{platform_str}{format_str}"
166
 
167
  #print(f"Round {round_num} query:\n{query}")
168