Bug Fix

Shiva Chilukamari · Shiva Chilukamari · commit 460fe422e973 · 2025-10-21T14:01:25.000+05:30
diff --git a/sam-vit-base/Olive/model_patches.py b/sam-vit-base/Olive/model_patches.py
@@ -301,7 +301,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, attn_weights
+        return hidden_states
 
 
 class ModSamModel(nn.Module):
