Update dependencyvit.py

huggingface · Dec 1, 2024 · f32418c · f32418c
1 parent 3c07131
commit f32418c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/timm/models/dependencyvit.py b/timm/models/dependencyvit.py
@@ -120,7 +120,7 @@ def forward(self, x: torch.Tensor, m: torch.Tensor) -> Tuple[torch.Tensor, torch
         #prune_mask = attn.sum(1).abs().sum(-1)
         #prune_mask = attn.abs().sum((1, -1))
         #prune_mask = m.reshape(B, N)
-        purne_mask = m.detach().reshape(B, N)
+        prune_mask = m.detach().reshape(B, N)
 
         x = self.proj(x)
         x = self.proj_drop(x)