Apply suggestions from code review

Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
This commit is contained in:
Younes B
2025-07-08 10:43:50 +04:00
committed by GitHub
parent 8555ee8b2c
commit 2dee7cf964

View File

@@ -289,7 +289,7 @@ class MODEL_ARCH(IntEnum):
LLAMA4 = auto() LLAMA4 = auto()
DECI = auto() DECI = auto()
FALCON = auto() FALCON = auto()
FALCON_H1 = auto() FALCON_H1 = auto()
BAICHUAN = auto() BAICHUAN = auto()
GROK = auto() GROK = auto()
GPT2 = auto() GPT2 = auto()
@@ -662,7 +662,7 @@ MODEL_ARCH_NAMES: dict[MODEL_ARCH, str] = {
MODEL_ARCH.DOTS1: "dots1", MODEL_ARCH.DOTS1: "dots1",
MODEL_ARCH.ARCEE: "arcee", MODEL_ARCH.ARCEE: "arcee",
MODEL_ARCH.ERNIE4_5: "ernie4_5", MODEL_ARCH.ERNIE4_5: "ernie4_5",
MODEL_ARCH.FALCON_H1: "falcon_h1", MODEL_ARCH.FALCON_H1: "falcon_h1",
} }
VISION_PROJECTOR_TYPE_NAMES: dict[VISION_PROJECTOR_TYPE, str] = { VISION_PROJECTOR_TYPE_NAMES: dict[VISION_PROJECTOR_TYPE, str] = {