CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chroma

Browse Source
This commit is contained in:
Haoming 2026-02-19 11:34:07 +08:00
parent a9a6b87a7e
commit 2b97de706c
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backend/nn/chroma.py
View File

@ -11,7 +11,6 @@ from backend.nn.flux import (
EmbedND,
MLPEmbedder,
ModulationOut,
RMSNorm,
SingleStreamBlock,
timestep_embedding,
)
@ -33,7 +32,7 @@ class Approximator(nn.Module):
super().__init__()
self.in_proj = nn.Linear(in_dim, hidden_dim, bias=True)
self.layers = nn.ModuleList([MLPEmbedder(hidden_dim, hidden_dim) for _ in range(n_layers)])
self.norms = nn.ModuleList([RMSNorm(hidden_dim) for _ in range(n_layers)])
self.norms = nn.ModuleList([nn.RMSNorm(hidden_dim) for _ in range(n_layers)])
self.out_proj = nn.Linear(hidden_dim, out_dim)
@property