Update src/transformers/models/vitpose/modeling_vitpose.py

Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>
NielsRogge · Oct 6, 2024 · 8cb5f9c · 8cb5f9c
1 parent fee2582
commit 8cb5f9c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/vitpose/modeling_vitpose.py b/src/transformers/models/vitpose/modeling_vitpose.py
@@ -204,7 +204,7 @@ class VitPoseClassicDecoder(nn.Module):
  turning the feature maps into heatmaps.
  """
 
- def __init__(self, config):
+ def __init__(self, config: VitPoseConfig):
  super().__init__()
 
  self.deconv1 = nn.ConvTranspose2d(