mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
Fix std initialization in Idefics variants (#37100)
* Nit 😅
* Another one
* fix
* run ci
* revert change
This commit is contained in:
parent
786d9c5ed9
commit
4b13a02920
@ -604,9 +604,9 @@ class Idefics2PreTrainedModel(PreTrainedModel):
|
||||
|
||||
def _init_weights(self, module):
|
||||
std = (
|
||||
self.config.text_config.initializer_range
|
||||
self.config.initializer_range
|
||||
if hasattr(self.config, "initializer_range")
|
||||
else self.config.text_config.initializer_range
|
||||
else self.config.get_text_config().initializer_range
|
||||
)
|
||||
|
||||
if hasattr(module, "class_embedding"):
|
||||
|
@ -625,9 +625,9 @@ class Idefics3PreTrainedModel(PreTrainedModel):
|
||||
# Copied from transformers.models.idefics2.modeling_idefics2.Idefics2PreTrainedModel._init_weights
|
||||
def _init_weights(self, module):
|
||||
std = (
|
||||
self.config.text_config.initializer_range
|
||||
self.config.initializer_range
|
||||
if hasattr(self.config, "initializer_range")
|
||||
else self.config.text_config.initializer_range
|
||||
else self.config.get_text_config().initializer_range
|
||||
)
|
||||
|
||||
if hasattr(module, "class_embedding"):
|
||||
|
@ -85,9 +85,9 @@ class SmolVLMPreTrainedModel(PreTrainedModel):
|
||||
|
||||
def _init_weights(self, module):
|
||||
std = (
|
||||
self.config.text_config.initializer_range
|
||||
self.config.initializer_range
|
||||
if hasattr(self.config, "initializer_range")
|
||||
else self.config.text_config.initializer_range
|
||||
else self.config.get_text_config().initializer_range
|
||||
)
|
||||
|
||||
if hasattr(module, "class_embedding"):
|
||||
|
Loading…
Reference in New Issue
Block a user