riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

more fixes?

Browse Source
This commit is contained in:
Arthur 2025-07-01 16:29:15 +02:00
parent e2973440d1
commit 0c9f6de0fd
3 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/janus/modeling_janus.py
View File

@ -359,7 +359,7 @@ class JanusVisionAttention(nn.Module):
output = self.projection_layer(attn_output)
output = self.projection_dropout(output)
return output
return output, attn_weights
class JanusVisionMLP(nn.Module):

2
src/transformers/models/janus/modular_janus.py
View File

@ -529,7 +529,7 @@ class JanusVisionAttention(nn.Module):
output = self.projection_layer(attn_output)
output = self.projection_dropout(output)
return output
return output, attn_weights
class JanusVisionMLP(nn.Module):

5
src/transformers/utils/generic.py
View File

@ -952,10 +952,7 @@ def can_return_tuple(func):
@wraps(func)
def wrapper(self, *args, **kwargs):
return_dict = self.config.use_return_dict if hasattr(self, "config") else True
if "return_dict" in kwargs:
return_dict = kwargs.get("return_dict", self.config.use_return_dict)
if "return_dict" in kwargs:
kwargs["return_dict"] = True
return_dict = kwargs.pop("return_dict", self.config.use_return_dict)
output = func(self, *args, **kwargs)
if "return_dict" in kwargs and return_dict is False: