MCPcopy Create free account
hub / github.com/TencentARC/BrushNet / __init__

Method __init__

src/diffusers/models/autoencoders/vae.py:362–370  ·  view source on GitHub ↗
(
        self,
        in_channels: int,
        out_channels: int,
    )

Source from the content-addressed store, hash-verified

360 """
361
362 def __init__(
363 self,
364 in_channels: int,
365 out_channels: int,
366 ) -> None:
367 super().__init__()
368 self.in_channels = in_channels
369 self.out_channels = out_channels
370 self.deconv = nn.ConvTranspose2d(in_channels, out_channels, kernel_size=4, stride=2, padding=1)
371
372 def forward(self, x: torch.FloatTensor) -> torch.FloatTensor:
373 r"""The forward method of the `UpSample` class."""

Callers

nothing calls this directly

Calls 1

__init__Method · 0.45

Tested by

no test coverage detected