Update modeling_motif.py
Browse files- 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_|>
|