Skip to content

Commit ab0ff0f

Browse files
committed
also add to the llm_tensor_names
1 parent 9a49ae9 commit ab0ff0f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/llama-arch.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -778,6 +778,7 @@ static const std::map<llm_arch, std::map<llm_tensor, const char *>> LLM_TENSOR_N
778778
{
779779
{ LLM_TENSOR_TOKEN_EMBD, "token_embd" },
780780
{ LLM_TENSOR_OUTPUT_NORM, "output_norm" },
781+
{ LLM_TENSOR_OUTPUT, "output" },
781782
{ LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
782783
{ LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
783784
{ LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },

0 commit comments

Comments
 (0)