diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index 1c9f223eb..4ace13eb6 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -1978,6 +1978,7 @@ class Phi2Model(Model): self.gguf_writer.add_file_type(self.ftype) self.gguf_writer.add_add_bos_token(False) + @Model.register("Phi3ForCausalLM") class Phi3MiniModel(Model): model_arch = gguf.MODEL_ARCH.PHI3 @@ -2062,6 +2063,7 @@ class Phi3MiniModel(Model): self.gguf_writer.add_rope_dimension_count(int(rot_pct * n_embd) // n_head) self.gguf_writer.add_file_type(self.ftype) + @Model.register("PlamoForCausalLM") class PlamoModel(Model): model_arch = gguf.MODEL_ARCH.PLAMO @@ -2885,5 +2887,6 @@ def main() -> None: print(f"Model successfully exported to '{fname_out}'") + if __name__ == '__main__': main()