Wendy commited on
Commit
5367a16
·
verified ·
1 Parent(s): 0d409d1

Upload cogagent_infer_batch.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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",