Upload cogagent_infer_batch.py with huggingface_hub
Browse files- 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 =
|
|
|
|
|
|
|
|
|
|
| 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 |
|