Skip to content

Commit

Permalink
add ftype
Browse files Browse the repository at this point in the history
  • Loading branch information
mscheong01 committed Aug 16, 2024
1 parent 78c6e6c commit 98ad475
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions convert_hf_to_gguf.py
Original file line number Diff line number Diff line change
Expand Up @@ -3812,6 +3812,7 @@ def set_gguf_parameters(self):
self.gguf_writer.add_layer_norm_rms_eps(layer_norm_eps)
self.gguf_writer.add_feed_forward_length(intermediate_size)
self.gguf_writer.add_block_count(num_layers)
self.gguf_writer.add_file_type(self.ftype)

if (rope_theta := self.hparams.get("rope_theta")) is not None:
self.gguf_writer.add_rope_freq_base(rope_theta)
Expand Down

0 comments on commit 98ad475

Please sign in to comment.