mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-01 18:51:14 +06:00
test this
This commit is contained in:
parent
780141ca52
commit
3c0c56b84d
@ -954,7 +954,6 @@ def can_return_tuple(func):
|
|||||||
return_dict = self.config.use_return_dict if hasattr(self, "config") else True
|
return_dict = self.config.use_return_dict if hasattr(self, "config") else True
|
||||||
if "return_dict" in kwargs:
|
if "return_dict" in kwargs:
|
||||||
return_dict = kwargs.get("return_dict", self.config.use_return_dict)
|
return_dict = kwargs.get("return_dict", self.config.use_return_dict)
|
||||||
kwargs["return_dict"] = True
|
|
||||||
output = func(self, *args, **kwargs)
|
output = func(self, *args, **kwargs)
|
||||||
|
|
||||||
if "return_dict" in kwargs and return_dict is False:
|
if "return_dict" in kwargs and return_dict is False:
|
||||||
|
Loading…
Reference in New Issue
Block a user