fix multimodal model image_mode = 'CMYK' (#697)

modelscope · Apr 14, 2024 · 0a5c39b · 0a5c39b
1 parent 24e6b11
commit 0a5c39b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/swift/llm/utils/template.py b/swift/llm/utils/template.py
@@ -580,7 +580,7 @@ def _read_from_path(
             image = Image.open(img_path)
     else:
         image = img_path
-    if image.mode in {'L', 'RGBA'}:
+    if image.mode != 'RGB':
         image = image.convert('RGB')
     return image