Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/pipelines/base.py
View file @ 91472cf5
...@@ -896,7 +896,7 @@ class Pipeline(_ScikitCompat, PushToHubMixin): ...@@ -896,7 +896,7 @@ class Pipeline(_ScikitCompat, PushToHubMixin):
and self.tokenizer.pad_token_id is not None and self.tokenizer.pad_token_id is not None
and self.model.generation_config.pad_token_id is 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.call_count = 0
self._batch_size = kwargs.pop("batch_size", None) 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