Update modeling_super_linear.py

modeling_super_linear.py

@@ -448,9 +448,14 @@ class superLinear(nn.Module):
 
 
     def forward(self, x_enc, x_mark_enc=None, x_dec=None, x_mark_dec=None, mask=None, freq=[None], get_prob=False):
-        print(x_enc.shape)
-        x = x_enc.permute(0, 2, 1)
-        B, V, L = x.shape
+        if len(x_enc.shape) > 2:
+            x = x_enc.permute(0, 2, 1)
+            B, V, L = x.shape
+        else:
+            x = x_enc
+            B, L = x.shape
+            V    = 1
+            
         x = x.reshape(B * V, L)
         
         expert_probs = None