mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-23 17:56:07 +00:00
fix(policy): xvla forced_bos_token missing
This commit is contained in:
@@ -276,6 +276,8 @@ class Florence2LanguageConfig(PretrainedConfig):
|
|||||||
)
|
)
|
||||||
|
|
||||||
# ensure backward compatibility for BART CNN models
|
# ensure backward compatibility for BART CNN models
|
||||||
|
if not hasattr(self, "forced_bos_token_id"):
|
||||||
|
self.forced_bos_token_id = None
|
||||||
if self.forced_bos_token_id is None and kwargs.get("force_bos_token_to_be_generated", False):
|
if self.forced_bos_token_id is None and kwargs.get("force_bos_token_to_be_generated", False):
|
||||||
self.forced_bos_token_id = self.bos_token_id
|
self.forced_bos_token_id = self.bos_token_id
|
||||||
warnings.warn(
|
warnings.warn(
|
||||||
|
|||||||
Reference in New Issue
Block a user