|
22 | 22 |
|
23 | 23 | from ..pytorch_utils import prune_linear_layer |
24 | 24 | from ..utils import ModelOutput, is_sklearn_available |
25 | | -from ..utils.deprecation import deprecate_kwarg |
26 | 25 | from .configuration_utils import GenerationConfig |
27 | 26 | from .logits_process import LogitsProcessorList, MinLengthLogitsProcessor, SuppressTokensLogitsProcessor |
28 | 27 |
|
@@ -1250,21 +1249,11 @@ class SinglePositionMultiTokenCandidateGenerator(CandidateGenerator): |
1250 | 1249 | "return_shared_kv_states": True, |
1251 | 1250 | } |
1252 | 1251 |
|
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") |
1258 | 1252 | def __init__( |
1259 | 1253 | self, |
1260 | 1254 | assistant_model: "PreTrainedModel", |
1261 | 1255 | target_model_input_embeddings: nn.Embedding, |
1262 | 1256 | 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, |
1268 | 1257 | **kwargs, |
1269 | 1258 | ): |
1270 | 1259 | if ( |
|
0 commit comments