Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -1
modeling_super_linear.py
CHANGED
|
@@ -600,7 +600,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
|
|
| 600 |
|
| 601 |
# backbone returns (B, pred_len, C)
|
| 602 |
preds = self.backbone(x_enc)
|
| 603 |
-
preds = self.revin_layer(preds, 'denorm')
|
| 604 |
return CausalLMOutputWithCrossAttentions(loss=None,logits=preds,past_key_values=None,hidden_states=None,attentions=None,)
|
| 605 |
|
| 606 |
|
|
|
|
| 600 |
|
| 601 |
# backbone returns (B, pred_len, C)
|
| 602 |
preds = self.backbone(x_enc)
|
| 603 |
+
#preds = self.revin_layer(preds, 'denorm')
|
| 604 |
return CausalLMOutputWithCrossAttentions(loss=None,logits=preds,past_key_values=None,hidden_states=None,attentions=None,)
|
| 605 |
|
| 606 |
|