diff --git a/docs/chapter5/第五章 动手搭建大模型.md b/docs/chapter5/第五章 动手搭建大模型.md index 9899de8..128cfd9 100644 --- a/docs/chapter5/第五章 动手搭建大模型.md +++ b/docs/chapter5/第五章 动手搭建大模型.md @@ -104,7 +104,7 @@ output = norm(x) print(output.shape) out: -orch.Size([1, 50, 768]) +torch.Size([1, 50, 768]) ``` ### 5.1.3 构建 LLaMA2 Attention