chore(代码变更): 1. 清理无用代码 2. 更新依赖库 3. 优化项目结构 4. 修复小型bug 5. 增加注释以提高可读性

2026-07-08 15:07:21 +08:00
parent d88afb5fe7
commit 3f952c5fe9
29 changed files with 654718 additions and 2587 deletions
@@ -4,12 +4,12 @@ use burn::tensor::backend::Backend;
 use burn::tensor::{Float, Tensor};

 #[derive(Module, Debug)]
-pub struct RmsNorm<B: Backend> {
+pub struct RMSNorm<B: Backend> {
    pub weight: Param<Tensor<B, 1, Float>>,
    pub eps: f64,
 }

-impl<B: Backend> RmsNorm<B> {
+impl<B: Backend> RMSNorm<B> {
    pub fn new(dim: usize, eps: f64, device: &B::Device) -> Self {
        let weight = Initializer::Ones.init([dim], device);
        Self {