Skip to content

GitLab

Unverified Commit 28e2887a authored by NielsRogge's avatar NielsRogge Committed by GitHub
Browse files

⚠️ [VitDet] Fix test (#27832) 

Address test
parent b242d0f2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/vitdet/modeling_vitdet.py
View file @ 28e2887a
...@@ -815,8 +815,8 @@ class VitDetBackbone(VitDetPreTrainedModel, BackboneMixin): ...@@ -815,8 +815,8 @@ class VitDetBackbone(VitDetPreTrainedModel, BackboneMixin):
def forward( def forward(
self, self,
pixel_values: torch.Tensor, pixel_values: torch.Tensor,
output_attentions: Optional[bool] = None,
output_hidden_states: Optional[bool] = None, output_hidden_states: Optional[bool] = None,
output_attentions: Optional[bool] = None,
return_dict: Optional[bool] = None, return_dict: Optional[bool] = None,
) -> BackboneOutput: ) -> BackboneOutput:
""" """
......
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