eunhwanpark-motiftech commited on
Commit
7ebd625
·
verified ·
1 Parent(s): 22c1361

Update modeling_motif.py

Browse files
Files changed (1) hide show
  1. modeling_motif.py +0 -18
modeling_motif.py CHANGED
@@ -1200,24 +1200,6 @@ class MotifModel(MotifPreTrainedModel):
1200
  self.gradient_checkpointing = False
1201
  self.post_init()
1202
 
1203
- self.use_pipeline = config.use_pipeline
1204
- if self.use_pipeline:
1205
- logger.info('use reinforced pp..')
1206
- if config.num_stages==2:
1207
- ### moe version
1208
- if config.decontam_attn:
1209
- self.split_layers = [15]
1210
- else:
1211
- if num_hidden_layers == 32:
1212
- self.split_layers = [14] # 14: 15,17 # 13: 14:18
1213
- else:
1214
- self.split_layers = [6]
1215
- elif config.num_stages==3:
1216
- self.split_layers = [9,20] ## 10, 11, 11
1217
- else:
1218
- self.split_layers = [6,15,24] #7(0,7),9(6,15),9(15,24),7(24,31)
1219
- logger.info(f' check the split layers (moe): {self.split_layers}')
1220
-
1221
  self.scale_emb = 1
1222
 
1223
  # Reparameterization <|_1_|>
 
1200
  self.gradient_checkpointing = False
1201
  self.post_init()
1202
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1203
  self.scale_emb = 1
1204
 
1205
  # Reparameterization <|_1_|>