From f32418cefb174943704f823018fcc3e1f8891ded Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Wed, 27 Dec 2023 04:42:08 -0800
Subject: [PATCH] Update dependencyvit.py

---
 timm/models/dependencyvit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/timm/models/dependencyvit.py b/timm/models/dependencyvit.py
index 45c1d3d24d..92fd8612a0 100644
--- 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)