xinyun/ComfyUI-MochiWrapper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tweak

Browse Source
This commit is contained in:
kijai 2024-10-25 20:04:01 +03:00
parent 2e22529c99
commit e66735527c
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mochi_preview/dit/joint_model/mod_rmsnorm.py
View File

@ -18,6 +18,5 @@ class ModulatedRMSNorm(torch.autograd.Function):
return x_modulated.type_as(x) return x_modulated.type_as(x)
@torch.compiler.disable()
def modulated_rmsnorm(x, scale, eps=1e-6): def modulated_rmsnorm(x, scale, eps=1e-6):
return ModulatedRMSNorm.apply(x, scale, eps) return ModulatedRMSNorm.apply(x, scale, eps)