Public/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2026-09-18 08:44:49 +02:00
Code Issues Packages Projects Releases Wiki Activity

convert: add @ModelBase.example

Browse Source
This commit is contained in:
Xuan Son Nguyen
2026-08-17 02:31:10 +02:00
parent 4df29be4f4
commit 71c7866a16
84 changed files with 231 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
conversion/starcoder.py
+2
View File
@@ -4,6 +4,7 @@ from .base import ModelBase, TextModel, gguf
@ModelBase.register("GPTBigCodeForCausalLM")
@ModelBase.example("bigcode/gpt_bigcode-santacoder")
class StarCoderModel(TextModel):
model_arch = gguf.MODEL_ARCH.STARCODER
@@ -19,5 +20,6 @@ class StarCoderModel(TextModel):
@ModelBase.register("Starcoder2ForCausalLM")
@ModelBase.example("bigcode/starcoder2-3b")
class StarCoder2Model(TextModel):
model_arch = gguf.MODEL_ARCH.STARCODER2