admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 18:26:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: fix XVLA in transformers v5 (#3006)

Browse Source
This commit is contained in:
Jade Choghari
2026-02-24 12:29:48 +03:00
committed by GitHub
parent 1d048c7e2b
commit 8c95a71c94
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/xvla/modeling_florence2.py
+12 -7
View File
@@ -1951,7 +1951,10 @@ class Florence2Decoder(Florence2LanguagePreTrainedModel):
class Florence2LanguageModel(Florence2LanguagePreTrainedModel): class Florence2LanguageModel(Florence2LanguagePreTrainedModel):
_tied_weights_keys = ["encoder.embed_tokens.weight", "decoder.embed_tokens.weight"] _tied_weights_keys = {
"encoder.embed_tokens.weight": "shared.weight",
"decoder.embed_tokens.weight": "shared.weight",
}
def __init__(self, config: Florence2LanguageConfig): def __init__(self, config: Florence2LanguageConfig):
super().__init__(config) super().__init__(config)
@@ -2076,7 +2079,10 @@ class Florence2LanguageModel(Florence2LanguagePreTrainedModel):
class Florence2LanguageForConditionalGeneration(Florence2LanguagePreTrainedModel, GenerationMixin): class Florence2LanguageForConditionalGeneration(Florence2LanguagePreTrainedModel, GenerationMixin):
base_model_prefix = "model" base_model_prefix = "model"
_tied_weights_keys = ["encoder.embed_tokens.weight", "decoder.embed_tokens.weight", "lm_head.weight"] _tied_weights_keys = {
"model.encoder.embed_tokens.weight": "model.shared.weight",
"model.decoder.embed_tokens.weight": "model.shared.weight",
}
_keys_to_ignore_on_load_missing = ["final_logits_bias"] _keys_to_ignore_on_load_missing = ["final_logits_bias"]
def __init__(self, config: Florence2LanguageConfig): def __init__(self, config: Florence2LanguageConfig):
@@ -2436,11 +2442,10 @@ FLORENCE2_INPUTS_DOCSTRING = r"""
FLORENCE2_START_DOCSTRING, FLORENCE2_START_DOCSTRING,
) )
class Florence2ForConditionalGeneration(Florence2PreTrainedModel): class Florence2ForConditionalGeneration(Florence2PreTrainedModel):
_tied_weights_keys = [ _tied_weights_keys = {
"language_model.encoder.embed_tokens.weight", "language_model.model.encoder.embed_tokens.weight": "language_model.model.shared.weight",
"language_model.decoder.embed_tokens.weight", "language_model.model.decoder.embed_tokens.weight": "language_model.model.shared.weight",
"language_model.lm_head.weight", }
]
def __init__(self, config: Florence2Config): def __init__(self, config: Florence2Config):
super().__init__(config) super().__init__(config)