bacow/happy-llm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 修正 5.1.2 中输出张量形状的错误

Browse Source
This commit is contained in:
KMnO4-zx
2025-06-12 09:08:38 +08:00
parent 74908262f1
commit 6a47afc997
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/chapter5/第五章 动手搭建大模型.md
View File

@@ -104,7 +104,7 @@ output = norm(x)
print(output.shape) print(output.shape)
out: out:
orch.Size([1, 50, 288]) orch.Size([1, 50, 768])
``` ```
### 5.1.3 构建 LLaMA2 Attention ### 5.1.3 构建 LLaMA2 Attention