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

Fix formatting issues

Browse source
This commit is contained in:
Yoshi Suhara 2024-08-08 09:18:59 -07:00
parent aa2f4a79fe
commit 45e9d164ac
3 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

1
convert_hf_to_gguf.py
View file

@ -3734,6 +3734,7 @@ class ChatGLMModel(Model):
name = name.removeprefix("transformer.") name = name.removeprefix("transformer.")
return [(self.map_tensor_name(name), data_torch)] return [(self.map_tensor_name(name), data_torch)]
@Model.register("NemotronForCausalLM") @Model.register("NemotronForCausalLM")
class NemotronModel(Model): class NemotronModel(Model):
model_arch = gguf.MODEL_ARCH.NEMOTRON model_arch = gguf.MODEL_ARCH.NEMOTRON