File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -655,6 +655,8 @@ def set_gguf_parameters(self):
655
655
self .gguf_writer .add_feed_forward_length (self .hparams ["intermediate_size" ])
656
656
self .gguf_writer .add_head_count (head_count )
657
657
self .gguf_writer .add_head_count_kv (head_count_kv )
658
+ # note: config provides rms norm but it is actually layer norm
659
+ # ref: https://huggingface.co/OrionStarAI/Orion-14B-Chat/blob/276a17221ce42beb45f66fac657a41540e71f4f5/modeling_orion.py#L570-L571
658
660
self .gguf_writer .add_layer_norm_eps (self .hparams ["rms_norm_eps" ])
659
661
660
662
def write_tensors (self ):
@@ -1031,7 +1033,6 @@ def set_gguf_parameters(self):
1031
1033
self .gguf_writer .add_head_count_kv (head_count_kv )
1032
1034
self .gguf_writer .add_rope_freq_base (self .hparams ["rope_theta" ])
1033
1035
self .gguf_writer .add_layer_norm_eps (self .hparams ["layer_norm_eps" ])
1034
- self .gguf_writer .add_layer_norm_rms_eps (self .hparams ["rms_norm_eps" ])
1035
1036
1036
1037
def set_vocab (self ):
1037
1038
self ._set_vocab_sentencepiece ()
You can’t perform that action at this time.
0 commit comments