Wendy
commited on
Upload cogagent_infer_batch.py with huggingface_hub
Browse files- cogagent_infer_batch.py +1 -1
cogagent_infer_batch.py
CHANGED
@@ -167,7 +167,7 @@ def main():
|
|
167 |
#print(f"Round {round_num} query:\n{query}")
|
168 |
|
169 |
inputs = tokenizer.apply_chat_template(
|
170 |
-
[{"role": "user", "image": image, "content": query}],
|
171 |
add_generation_prompt=True,
|
172 |
tokenize=True,
|
173 |
return_tensors="pt",
|
|
|
167 |
#print(f"Round {round_num} query:\n{query}")
|
168 |
|
169 |
inputs = tokenizer.apply_chat_template(
|
170 |
+
[{"role": "user", "image": **image, "content": **query}],
|
171 |
add_generation_prompt=True,
|
172 |
tokenize=True,
|
173 |
return_tensors="pt",
|