File tree Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -262,7 +262,7 @@ def get_tokenized_context(max_len):
262
262
# Stop conditions
263
263
264
264
sc = prompt_format .stop_conditions (tokenizer )
265
- sc = [x for x in sc if x ]
265
+ sc = [x for x in sc if x is not None ]
266
266
generator .set_stop_conditions (sc )
267
267
268
268
# ANSI color codes
Original file line number Diff line number Diff line change @@ -401,6 +401,7 @@ class Params:
401
401
"mlp_up" : ".mlp.c_fc" ,
402
402
"mlp_down" : ".mlp.c_proj" ,
403
403
"lm_head" : "model.embed_tokens" ,
404
+ "norm_eps" : "layer_norm_epsilon" ,
404
405
})
405
406
self .lm .mlp_act_func = "gelu"
406
407
self .lm .norm = "layernorm"
@@ -471,9 +472,9 @@ class Params:
471
472
"mlp_up" : ".block_sparse_moe.experts.*.v1" ,
472
473
"mlp_down" : ".block_sparse_moe.experts.*.w2" ,
473
474
"mlp_expert_gate" : ".block_sparse_moe.gate" ,
474
- "lm_head" : "model.embed_tokens" ,
475
475
"fused_qkv" : ".self_attn.Wqkv" ,
476
476
})
477
+ self .lm .norm = "layernorm"
477
478
self .lm .is_moe = True
478
479
479
480
# Phi3
You can’t perform that action at this time.
0 commit comments