AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

re-add super.__init__ for all PyTorch modules

Browse Source
This commit is contained in:
Patrick von Platen
2022-07-20 13:49:00 +00:00
parent ad9d252596
commit 919e27d357
3 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/diffusers/models/unet_conditional.py
View File

@@ -65,6 +65,7 @@ class UNetConditionalModel(ModelMixin, ConfigMixin):
center_input_sample=False,
resnet_num_groups=30,
):
super().__init__()
self.image_size = image_size
time_embed_dim = block_channels[0] * 4

1
src/diffusers/models/unet_unconditional.py
View File

@@ -61,6 +61,7 @@ class UNetUnconditionalModel(ModelMixin, ConfigMixin):
center_input_sample=False,
resnet_num_groups=32,
):
super().__init__()
self.image_size = image_size
time_embed_dim = block_channels[0] * 4

2
src/diffusers/models/vae.py
View File

@@ -400,6 +400,7 @@ class VQModel(ModelMixin, ConfigMixin):
resamp_with_conv=True,
give_pre_end=False,
):
super().__init__()
# pass init params to Encoder
self.encoder = Encoder(
@@ -477,6 +478,7 @@ class AutoencoderKL(ModelMixin, ConfigMixin):
resamp_with_conv=True,
give_pre_end=False,
):
super().__init__()
# pass init params to Encoder
self.encoder = Encoder(