diff --git a/lmdeploy/vl/model/xcomposer2.py b/lmdeploy/vl/model/xcomposer2.py index b93c784c8..188c8e9b1 100644 --- a/lmdeploy/vl/model/xcomposer2.py +++ b/lmdeploy/vl/model/xcomposer2.py @@ -304,8 +304,8 @@ def proc_messages(messages, chat_template, sequence_start, model_type): if n_images == 1: prefix_image_token, prompt = IMAGE_TOKEN, content[0] else: - prompt = '; '.join([ - f'Image{i+1} {IMAGE_TOKEN}' + prompt = ''.join([ + f'Image{i+1} {IMAGE_TOKEN}; ' for i in range(n_images) ]) + content[0] else: