mirror of
https://github.com/saymrwulf/transformers.git
synced 2026-05-14 20:58:08 +00:00
parent
ff56b8fbff
commit
dbfeffd7c9
1 changed files with 1 additions and 1 deletions
|
|
@ -102,7 +102,7 @@ def _sanitize_parameters(self, **kwargs):
|
|||
|
||||
postprocess_kwargs = {}
|
||||
if "top_k" in kwargs:
|
||||
preprocess_kwargs["top_k"] = kwargs["top_k"]
|
||||
postprocess_kwargs["top_k"] = kwargs["top_k"]
|
||||
return preprocess_kwargs, {}, postprocess_kwargs
|
||||
```
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue