mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-01 18:51:14 +06:00
parent
ff56b8fbff
commit
dbfeffd7c9
@ -102,7 +102,7 @@ def _sanitize_parameters(self, **kwargs):
|
|||||||
|
|
||||||
postprocess_kwargs = {}
|
postprocess_kwargs = {}
|
||||||
if "top_k" in 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
|
return preprocess_kwargs, {}, postprocess_kwargs
|
||||||
```
|
```
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user