mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-02 03:32:50 +08:00
[data] fix mm template (#7181)
Former-commit-id: be66df1f0211cd2d90eac3ab407dced653c9e443
This commit is contained in:
parent
002f58ef8e
commit
2b21c749c1
@ -1042,15 +1042,15 @@ class Qwen2VLPlugin(BasePlugin):
|
||||
image = super()._preprocess_image(image, **kwargs)
|
||||
if min(image.width, image.height) < 28:
|
||||
width, height = max(image.width, 28), max(image.height, 28)
|
||||
image = image.resize((width, height), resample=Image.Resampling.NEAREST)
|
||||
image = image.resize((width, height))
|
||||
|
||||
if image.width / image.height > 200:
|
||||
width, height = image.height * 180, image.height
|
||||
image = image.resize((width, height), resample=Image.Resampling.NEAREST)
|
||||
image = image.resize((width, height))
|
||||
|
||||
if image.height / image.width > 200:
|
||||
width, height = image.width, image.width * 180
|
||||
image = image.resize((width, height), resample=Image.Resampling.NEAREST)
|
||||
image = image.resize((width, height))
|
||||
|
||||
return image
|
||||
|
||||
|
@ -1268,9 +1268,17 @@ register_template(
|
||||
)
|
||||
|
||||
|
||||
# copied from gemma template
|
||||
register_template(
|
||||
name="paligemma",
|
||||
format_user=StringFormatter(slots=["{{content}}\n"]),
|
||||
format_prefix=EmptyFormatter(slots=[{"bos_token"}]),
|
||||
mm_plugin=get_mm_plugin(name="paligemma", image_token="<image>"),
|
||||
)
|
||||
|
||||
|
||||
# copied from gemma template
|
||||
register_template(
|
||||
name="paligemma_chat",
|
||||
format_user=StringFormatter(slots=["<start_of_turn>user\n{{content}}<end_of_turn>\n<start_of_turn>model\n"]),
|
||||
format_assistant=StringFormatter(slots=["{{content}}<end_of_turn>\n"]),
|
||||
format_observation=StringFormatter(
|
||||
|
Loading…
x
Reference in New Issue
Block a user