Update modeling_super_linear.py

modeling_super_linear.py

@@ -548,9 +548,9 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
         
         # backbone expects (B, C, L)
         x_enc = inputs_embeds
-        print(x_enc.shape)
+        #print(x_enc.shape)
         if x_enc.shape[1] < 512:
-            if len(x_enc) == 2:
+            if len(x_enc.shape) == 2:
                 x_enc = F.interpolate(x_enc.unsqueeze(0),size = 512,mode="linear", align_corners=False)
             else:
                 x_enc = F.interpolate(x_enc,size = 512,mode="linear", align_corners=False)