SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Remove deprecation warning for num_logits_to_keep (#37149)

Browse Source 
* remove everything

* style
This commit is contained in:
Cyril Vallez
2025-04-14 19:08:45 +02:00
committed by GitHub
parent 20ceaca228
commit 8ab296501a
56 changed files with 23 additions and 119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/models/aria/modeling_aria.py
View File

@@ -41,7 +41,6 @@ from ...utils import (
logging,
replace_return_docstrings,
)
from ...utils.deprecation import deprecate_kwarg
from ...utils.import_utils import is_torch_available
from ..auto import AutoModel, AutoModelForCausalLM
from .configuration_aria import AriaConfig, AriaTextConfig
@@ -1160,7 +1159,6 @@ class AriaTextForCausalLM(AriaTextPreTrainedModel, GenerationMixin):
return self.model
@can_return_tuple
@deprecate_kwarg("num_logits_to_keep", version="4.50", new_name="logits_to_keep")
@add_start_docstrings_to_model_forward(ARIA_TEXT_INPUTS_DOCSTRING)
@replace_return_docstrings(output_type=CausalLMOutputWithPast, config_class=_CONFIG_FOR_DOC)
def forward(
@@ -1417,7 +1415,6 @@ class AriaForConditionalGeneration(AriaPreTrainedModel, GenerationMixin):
return image_features
@can_return_tuple
@deprecate_kwarg("num_logits_to_keep", version="4.50", new_name="logits_to_keep")
@add_start_docstrings_to_model_forward(ARIA_INPUTS_DOCSTRING)
@replace_return_docstrings(output_type=AriaCausalLMOutputWithPast, config_class=AriaConfig)
def forward(

2
src/transformers/models/aria/modular_aria.py
View File

@@ -49,7 +49,6 @@ from ...utils import (
logging,
replace_return_docstrings,
)
from ...utils.deprecation import deprecate_kwarg
from ...utils.import_utils import is_torch_available
from ..auto import CONFIG_MAPPING, AutoConfig, AutoModel, AutoModelForCausalLM, AutoTokenizer
from ..llama.configuration_llama import LlamaConfig
@@ -1452,7 +1451,6 @@ class AriaForConditionalGeneration(AriaPreTrainedModel, GenerationMixin):
return image_features
@can_return_tuple
@deprecate_kwarg("num_logits_to_keep", version="4.50", new_name="logits_to_keep")
@add_start_docstrings_to_model_forward(ARIA_INPUTS_DOCSTRING)
@replace_return_docstrings(output_type=AriaCausalLMOutputWithPast, config_class=AriaConfig)
def forward(