Skip to content

Commit f28af0d

Browse files
authored
gguf-py: Support 01.AI Yi models (#3943)
1 parent d9b33fe commit f28af0d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

gguf-py/gguf/gguf.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -393,6 +393,7 @@ class TensorNameMap:
393393
"layers.{bid}.attention_norm", # llama-pth
394394
"encoder.layer.{bid}.attention.output.LayerNorm", # bert
395395
"language_model.encoder.layers.{bid}.input_layernorm", # persimmon
396+
"model.layers.{bid}.ln1", # yi
396397
),
397398

398399
# Attention norm 2
@@ -464,6 +465,7 @@ class TensorNameMap:
464465
"layers.{bid}.ffn_norm", # llama-pth
465466
"encoder.layer.{bid}.output.LayerNorm", # bert
466467
"language_model.encoder.layers.{bid}.post_attention_layernorm", # persimmon
468+
"model.layers.{bid}.ln2", # yi
467469
),
468470

469471
# Feed-forward up

0 commit comments

Comments
 (0)