Skip to content

Commit 25ae389

Browse files
committed
Merge branch 'DC-AE' of github.com:lawrence-cj/diffusers into DC-AE
2 parents 883bcf4 + 1752afd commit 25ae389

File tree

1 file changed

+0
-10
lines changed
  • src/diffusers/models/autoencoders

1 file changed

+0
-10
lines changed

src/diffusers/models/autoencoders/dc_ae.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -41,16 +41,6 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
4141
return x
4242

4343

44-
def build_norm(name: Optional[str]="bn2d", num_features: Optional[int]=None) -> Optional[nn.Module]:
45-
if name is None:
46-
norm = None
47-
elif name == "rms2d":
48-
norm = RMSNorm2d(normalized_shape=num_features)
49-
elif name == "bn2d":
50-
norm = BatchNorm2d(num_features=num_features)
51-
else:
52-
raise ValueError(f"norm {name} is not supported")
53-
return norm
5444

5545

5646
class ConvLayer(nn.Module):

0 commit comments

Comments
 (0)