Merge branch 'refactor-vl' into refactor-multimodal

InternLM · Nov 20, 2024 · a2bbec5 · a2bbec5
2 parents 4124c56 + 9e18529
commit a2bbec5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lmdeploy/vl/model/internvl.py b/lmdeploy/vl/model/internvl.py
@@ -245,7 +245,7 @@ def proc_messages(cls, messages, chat_template, sequence_start):
                 prompt = f'<img>{IMAGE_TOKEN * n_images}</img>\n' + prompt
             else:
                 pass
-            prompt_messages.append(dict(role='user', content=content))
+            prompt_messages.append(dict(role='user', content=prompt))
         prompt = chat_template.messages2prompt(prompt_messages, sequence_start)
         return prompt, IMAGE_TOKEN