Update modeling_super_linear.py
Browse files- modeling_super_linear.py +2 -4
modeling_super_linear.py
CHANGED
|
@@ -337,8 +337,6 @@ class superLinear(nn.Module):
|
|
| 337 |
super(superLinear, self).__init__()
|
| 338 |
|
| 339 |
self.configs = configs
|
| 340 |
-
print("welcome super linear")
|
| 341 |
-
print(configs)
|
| 342 |
self.pred_len = configs.pred_len
|
| 343 |
self.seq_len = configs.seq_len
|
| 344 |
self.inf_pred_len = configs.inf_pred_len
|
|
@@ -496,8 +494,8 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
|
|
| 496 |
|
| 497 |
|
| 498 |
# the backbone keeps its own Config dataclass, so build one on‑the‑fly:
|
| 499 |
-
print("SuperLinearForCausalLM")
|
| 500 |
-
print(config)
|
| 501 |
backbone_cfg = type("Cfg", (), config.to_dict())()
|
| 502 |
self.backbone = superLinear(backbone_cfg)
|
| 503 |
|
|
|
|
| 337 |
super(superLinear, self).__init__()
|
| 338 |
|
| 339 |
self.configs = configs
|
|
|
|
|
|
|
| 340 |
self.pred_len = configs.pred_len
|
| 341 |
self.seq_len = configs.seq_len
|
| 342 |
self.inf_pred_len = configs.inf_pred_len
|
|
|
|
| 494 |
|
| 495 |
|
| 496 |
# the backbone keeps its own Config dataclass, so build one on‑the‑fly:
|
| 497 |
+
#print("SuperLinearForCausalLM")
|
| 498 |
+
#print(config)
|
| 499 |
backbone_cfg = type("Cfg", (), config.to_dict())()
|
| 500 |
self.backbone = superLinear(backbone_cfg)
|
| 501 |
|