Bug fix

Shiva Chilukamari · Shiva Chilukamari · commit 1e02d163be8c · 2025-11-19T10:09:58.000+05:30
diff --git a/sam-vit-base/QNN/model_patches.py b/sam-vit-base/QNN/model_patches.py
@@ -296,7 +296,7 @@ def forward(
         hidden_states = residual + hidden_states
         layernorm_output = self.model.layer_norm2(hidden_states)
         hidden_states = hidden_states + self.mlp(layernorm_output)
-        return hidden_states
+        return hidden_states, attn_weights
 
 
 class ModSamModel(nn.Module):
