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

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