xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-02 01:57:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Replace manual x * sigmoid(x) with torch silu in VAE nonlinearity

Browse Source
This commit is contained in:
chaObserv 2025-07-25 22:45:58 +08:00
parent b850d9a8bb
commit 0a91d0f598
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/ldm/cosmos/cosmos_tokenizer/utils.py
View File

@ -58,7 +58,8 @@ def is_odd(n: int) -> bool:
def nonlinearity(x):
return x * torch.sigmoid(x)
# x * sigmoid(x)
return torch.nn.functional.silu(x)
def Normalize(in_channels, num_groups=32):

2
comfy/ldm/modules/diffusionmodules/model.py
View File

@ -36,7 +36,7 @@ def get_timestep_embedding(timesteps, embedding_dim):
def nonlinearity(x):
# swish
return x*torch.sigmoid(x)
return torch.nn.functional.silu(x)
def Normalize(in_channels, num_groups=32):