vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

convert : fix lint checks

Browse source
This commit is contained in:
Georgi Gerganov 2024-04-24 09:45:35 +03:00
parent ae133e7fa6
commit 725cf63646
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
convert-hf-to-gguf.py
View file

@ -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()