Update modeling_super_linear.py

modeling_super_linear.py

@@ -596,14 +596,14 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
         # Add channel dimension if input is 2D
         size = len(x.shape)
         if size == 2:
-            x = x.unsqueeze(1)  # [B, 1, L]
-        
+            x = x.unsqueeze(1)  
         
+        print(target_len)
         upsample =  F.interpolate(x, size=target_len, mode='linear', align_corners=False) 
         
         # If input was 2D, remove the channel dimension
         if size == 2:
-            upsample = upsample.squeeze(-1)
+            upsample = upsample.squeeze(1)
 
         upsample = upsample.float()