Fix formatting issues

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

View file

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