From a2bd8b479e4364ab652d285f8d84491ecef19c61 Mon Sep 17 00:00:00 2001 From: Junli Liu <85859541+Ideal-ljl@users.noreply.github.com> Date: Thu, 9 Oct 2025 11:29:14 +0800 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 5164390..af9a656 100644 --- a/README.md +++ b/README.md @@ -149,13 +149,13 @@ messages = [ ] # 2. text generation [multimodal reasoning] -input_length = inputs["input_ids"].shape[1] inputs = processor.apply_chat_template( messages, tokenize=True, return_dict=True, return_tensors="pt" ).to("cuda") +input_length = inputs["input_ids"].shape[1] outputs = model.generate(**inputs, max_new_tokens=1024, return_dict_in_generate=True) generated_ids = outputs.sequences