Fix accidental removal

2024-04-13 21:15:15 -07:00 · 2024-04-13 21:15:15 -07:00 · 8dcd9978d2
commit 8dcd9978d2
parent 0ec53cfff7
1 changed files with 1 additions and 0 deletions
--- a/gguf-py/gguf/constants.py
+++ b/gguf-py/gguf/constants.py
@ -196,6 +196,7 @@ MODEL_ARCH_NAMES: dict[MODEL_ARCH, str] = {
    MODEL_ARCH.MAMBA:          "mamba",
    MODEL_ARCH.XVERSE:         "xverse",
    MODEL_ARCH.COMMAND_R:      "command-r",
+    MODEL_ARCH.DBRX:           "dbrx",
 }

 TENSOR_NAMES: dict[MODEL_TENSOR, str] = {