Unverified Commit 91472cf5 authored by Raushan Turganbay's avatar Raushan Turganbay Committed by GitHub
Browse files

Pipeline: fix `pad_token_id` again (#30338)

fix again
parent cd09a8df
......@@ -896,7 +896,7 @@ class Pipeline(_ScikitCompat, PushToHubMixin):
and self.tokenizer.pad_token_id is not None
and self.model.generation_config.pad_token_id is None
):
kwargs["pad_token_id"] = self.tokenizer.pad_token_id
self.model.generation_config.pad_token_id = self.tokenizer.pad_token_id
self.call_count = 0
self._batch_size = kwargs.pop("batch_size", None)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment