Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -1
modeling_super_linear.py
CHANGED
|
@@ -527,7 +527,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
|
|
| 527 |
# backbone returns (B, pred_len, C)
|
| 528 |
preds = self.backbone(x_enc)
|
| 529 |
print(F"preds shape: {preds.shape}")
|
| 530 |
-
preds = preds[0]
|
| 531 |
print(F"preds shape: {preds.shape}")
|
| 532 |
|
| 533 |
# if we keep continuous values, treat them as logits directly
|
|
|
|
| 527 |
# backbone returns (B, pred_len, C)
|
| 528 |
preds = self.backbone(x_enc)
|
| 529 |
print(F"preds shape: {preds.shape}")
|
| 530 |
+
#preds = preds[0]
|
| 531 |
print(F"preds shape: {preds.shape}")
|
| 532 |
|
| 533 |
# if we keep continuous values, treat them as logits directly
|