riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

update with setdefault

Browse Source
This commit is contained in:
raushan 2025-07-02 11:49:31 +02:00
parent 95d6438fc3
commit 343fd32dd6
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/transformers/models/smolvlm/processing_smolvlm.py
View File

@ -391,8 +391,6 @@ class SmolVLMProcessor(ProcessorMixin):
self,
conversation: Union[list[dict[str, str]], list[list[dict[str, str]]]],
chat_template: Optional[str] = None,
num_frames: Optional[int] = None,
fps: Optional[int] = None,
**kwargs: Unpack[AllKwargsForChatTemplate],
) -> str:
"""
@ -437,10 +435,10 @@ class SmolVLMProcessor(ProcessorMixin):
# re-assign to the correct default template for BC, if user is not requesting their own template
chat_template = DEFAULT_CHAT_TEMPLATE
num_frames = self.video_processor.num_frames if num_frames is None else num_frames
fps = self.video_processor.fps if fps is None else fps
kwargs.setdefault("num_frames", self.video_processor.num_frames)
kwargs.setdefault("fps", self.video_processor.fps)
return super().apply_chat_template(conversation, chat_template, num_frames=num_frames, fps=fps, **kwargs)
return super().apply_chat_template(conversation, chat_template, **kwargs)
__all__ = ["SmolVLMProcessor"]