We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 883bcf4 + 1752afd commit 25ae389Copy full SHA for 25ae389
src/diffusers/models/autoencoders/dc_ae.py
@@ -41,16 +41,6 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
41
return x
42
43
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
54
55
56
class ConvLayer(nn.Module):
0 commit comments