Skip to content
This repository was archived by the owner on Nov 12, 2025. It is now read-only.

Commit 5612735

Browse files
authored
Update README.md (#31)
1 parent 541b185 commit 5612735

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,13 +149,13 @@ messages = [
149149
]
150150

151151
# 2. text generation [multimodal reasoning]
152-
input_length = inputs["input_ids"].shape[1]
153152
inputs = processor.apply_chat_template(
154153
messages,
155154
tokenize=True,
156155
return_dict=True,
157156
return_tensors="pt"
158157
).to("cuda")
158+
input_length = inputs["input_ids"].shape[1]
159159

160160
outputs = model.generate(**inputs, max_new_tokens=1024, return_dict_in_generate=True)
161161
generated_ids = outputs.sequences

0 commit comments

Comments
 (0)