uvos/HyDecodeRepo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add profileing and small conv3d repoducer

Browse Source
This commit is contained in:
uvos
2025-02-12 16:29:43 +01:00
parent b279e2dfd7
commit 54a585f2b9
4 changed files with 67 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
vae/autoencoder_kl_causal_3d.py
View File

@ -115,6 +115,9 @@ class AutoencoderKLCausal3D(ModelMixin, ConfigMixin, FromOriginalVAEMixin):
self.post_quant_conv = nn.Conv3d(
latent_channels, latent_channels, kernel_size=1)
print(f"Conv3d: {2 * latent_channels}, {2 * latent_channels}, 1")
print(f"Conv3d: {latent_channels}, {latent_channels}, 1")
self.use_slicing = False
self.use_spatial_tiling = False
self.use_temporal_tiling = False

1
vae/unet_causal_3d_blocks.py
View File

@ -72,6 +72,7 @@ class CausalConv3d(nn.Module):
self.conv = nn.Conv3d(chan_in, chan_out, kernel_size,
stride=stride, dilation=dilation, **kwargs)
print(f"Conv3d: {chan_in}, {chan_out}, {kernel_size}, stride={stride}, dilation={dilation}")
def forward(self, x):
x = F.pad(x, self.time_causal_padding, mode=self.pad_mode)