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 7e1b315 commit eba9df6
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 @@ -3626,6 +3626,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 eba9df6

Please sign in to comment.