Skip to content

Commit 5044ab6

Browse files
committed
Set rope_theta
1 parent 5ff0500 commit 5044ab6

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

convert_hf_to_gguf.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9563,6 +9563,7 @@ def set_gguf_parameters(self):
95639563
self.gguf_writer.add_vocab_size(self.hparams["vocab_size"])
95649564
self.gguf_writer.add_shortconv_l_cache(self.hparams["conv_L_cache"])
95659565
self.gguf_writer.add_layer_norm_rms_eps(self.hparams["norm_eps"])
9566+
self.gguf_writer.add_rope_freq_base(self.hparams.get("rope_theta", 1000000))
95669567
self._add_feed_forward_length()
95679568

95689569
def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iterable[tuple[str, Tensor]]:

0 commit comments

Comments
 (0)