diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index aed84f1b2..8767c89a8 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -984,6 +984,7 @@ class QwenModel(Model): print(f"{new_name}, n_dims = {n_dims}, {old_dtype} --> {data.dtype}") self.gguf_writer.add_tensor(new_name, data) + class Phi2Model(Model): def set_gguf_parameters(self): block_count = self.hparams["n_layer"] @@ -999,6 +1000,7 @@ class Phi2Model(Model): self.gguf_writer.add_rope_dimension_count(self.hparams["rotary_dim"]) self.gguf_writer.add_file_type(self.ftype) + ###### CONVERSION LOGIC ######