Update the JanusPro-generation

swift/llm/template/base.py

@@ -227,13 +227,11 @@ def _replace_image_tags(inputs: StdTemplateInputs):
     def _replace_start_image_tags(inputs: StdTemplateInputs):
         # compat
         generate_mode = False
-        for message in inputs.messages:
-            content = message['content']
-            if not isinstance(content, str):
-                continue
-            if content.strip().endswith('<start-image>'):
-                generate_mode = True
-                message['content'] = re.sub('<start-image>', '', content).strip()  # remove the <start-image>
+        message = inputs.messages[-1]
+        content = message['content']
+        if message['role'] == 'user' and content.endswith('<start-image>'):
+            generate_mode = True
+            message['content'] = message['content'][:-len('<start-image>')]  # remove the <start-image>
         inputs.generate_mode = generate_mode
 
     def _rlhf_encode(self, inputs: StdTemplateInputs) -> Dict[str, Any]:

tests/test_align/test_template/test_gene.py

@@ -8,10 +8,7 @@
 
 def test_deepseek_janus_pro_gene():
     from swift.llm import infer_main, InferArguments
-    args = InferArguments(
-        # model='deepseek-ai/Janus-Pro-1B',
-        model='/mnt/nas1/.cache/modelscope/hub/deepseek-ai/Janus-Pro-1B',
-        infer_backend='pt')
+    args = InferArguments(model='deepseek-ai/Janus-Pro-1B', infer_backend='pt')
     infer_main(args)