razmars commited on
Commit
1955d35
·
verified ·
1 Parent(s): 932eafc

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. 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