Skip to content

Commit b7e80d2

Browse files
authored
Remove deprecation cycle for candidate generators (#50034)
remove
1 parent 455c9a0 commit b7e80d2

1 file changed

Lines changed: 0 additions & 11 deletions

File tree

‎src/transformers/generation/candidate_generator.py‎

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222

2323
from ..pytorch_utils import prune_linear_layer
2424
from ..utils import ModelOutput, is_sklearn_available
25-
from ..utils.deprecation import deprecate_kwarg
2625
from .configuration_utils import GenerationConfig
2726
from .logits_process import LogitsProcessorList, MinLengthLogitsProcessor, SuppressTokensLogitsProcessor
2827

@@ -1250,21 +1249,11 @@ class SinglePositionMultiTokenCandidateGenerator(CandidateGenerator):
12501249
"return_shared_kv_states": True,
12511250
}
12521251

1253-
@deprecate_kwarg("input_ids", version="5.20.0")
1254-
@deprecate_kwarg("model_kwargs", version="5.20.0")
1255-
@deprecate_kwarg("inputs_tensor", version="5.20.0")
1256-
@deprecate_kwarg("logits_processor", version="5.20.0")
1257-
@deprecate_kwarg("eos_token_id", version="5.20.0")
12581252
def __init__(
12591253
self,
12601254
assistant_model: "PreTrainedModel",
12611255
target_model_input_embeddings: nn.Embedding,
12621256
generation_config: "GenerationConfig",
1263-
input_ids: torch.LongTensor | None = None,
1264-
model_kwargs: dict | None = None,
1265-
inputs_tensor: torch.Tensor | None = None,
1266-
logits_processor: Optional["LogitsProcessorList"] = None,
1267-
eos_token_id: int | list[int] | torch.Tensor | None = None,
12681257
**kwargs,
12691258
):
12701259
if (

0 commit comments

Comments
 (0)