Fix: failing CI after #30568 (#30599)

* failiing CI * no let's keep it intil full deprecation in v4.42
huggingface · May 2, 2024 · 5cf3e6b · 5cf3e6b
1 parent c681b58
commit 5cf3e6b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py
@@ -1905,6 +1905,9 @@ def resize_token_embeddings(
  return model_embeds
 
  # Update base model and current model config
+ if hasattr(self.config, "text_config"):
+ self.config.text_config.vocab_size = model_embeds.weight.shape[0]
+ # TODO: to be removed after v4.42, config.vocab_size is deprecated for models that have a config.text_config
  self.config.vocab_size = model_embeds.weight.shape[0]
  self.vocab_size = model_embeds.weight.shape[0]