TdAI / llama.cpp /conversion /maincoder.py
tda45's picture
Upload folder using huggingface_hub (part 2)
c305fa5 verified
Raw
History Blame Contribute Delete
409 Bytes
from __future__ import annotations
from .base import ModelBase, TextModel, gguf
@ModelBase.register("MaincoderForCausalLM")
class MaincoderModel(TextModel):
model_arch = gguf.MODEL_ARCH.MAINCODER
def set_gguf_parameters(self):
super().set_gguf_parameters()
if (head_dim := self.hparams.get("head_dim")) is not None:
self.gguf_writer.add_rope_dimension_count(head_dim)