Skip to content

Commit 54956d2

Browse files
committed
vqvae
1 parent 16f4789 commit 54956d2

File tree

3 files changed

+3
-9
lines changed

3 files changed

+3
-9
lines changed

src/diffusers/pipelines/amused/pipeline_amused.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -67,9 +67,7 @@ def __init__(
6767
scheduler=scheduler,
6868
)
6969
self.vae_scale_factor = (
70-
2 ** (len(self.vqvae.config.block_out_channels) - 1)
71-
if hasattr(self, "vqvae") and self.vqvae is not None
72-
else 8
70+
2 ** (len(self.vqvae.config.block_out_channels) - 1) if getattr(self, "vqvae", None) else 8
7371
)
7472
self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor, do_normalize=False)
7573

src/diffusers/pipelines/amused/pipeline_amused_img2img.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -82,9 +82,7 @@ def __init__(
8282
scheduler=scheduler,
8383
)
8484
self.vae_scale_factor = (
85-
2 ** (len(self.vqvae.config.block_out_channels) - 1)
86-
if hasattr(self, "vqvae") and self.vqvae is not None
87-
else 8
85+
2 ** (len(self.vqvae.config.block_out_channels) - 1) if getattr(self, "vqvae", None) else 8
8886
)
8987
self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor, do_normalize=False)
9088

src/diffusers/pipelines/amused/pipeline_amused_inpaint.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -90,9 +90,7 @@ def __init__(
9090
scheduler=scheduler,
9191
)
9292
self.vae_scale_factor = (
93-
2 ** (len(self.vqvae.config.block_out_channels) - 1)
94-
if hasattr(self, "vqvae") and self.vqvae is not None
95-
else 8
93+
2 ** (len(self.vqvae.config.block_out_channels) - 1) if getattr(self, "vqvae", None) else 8
9694
)
9795
self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor, do_normalize=False)
9896
self.mask_processor = VaeImageProcessor(

0 commit comments

Comments
 (0)