make the vae can be None in __init__ of SanaPipeline

lawrence-cj · lawrence-cj · commit 6ab2d8d07792 · 2024-12-09T18:33:05.000+08:00
diff --git a/src/diffusers/pipelines/sana/pipeline_sana.py b/src/diffusers/pipelines/sana/pipeline_sana.py
@@ -167,7 +167,9 @@ def __init__(
             tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler
         )
 
-        self.vae_scale_factor = 2 ** (len(self.vae.config.encoder_width_list) - 1)
+        self.vae_scale_factor = (
+            2 ** (len(self.vae.config.encoder_width_list) - 1) if hasattr(self, "vae") and self.vae is not None else 32
+        )
         self.image_processor = PixArtImageProcessor(vae_scale_factor=self.vae_scale_factor)
 
     # Copied from diffusers.pipelines.pixart_alpha.pipeline_pixart_alpha.PixArtAlphaPipeline.encode_prompt with 120->300

Original file line number	Diff line number	Diff line change
`@@ -167,7 +167,9 @@ def __init__(`
`167`	`167`	`tokenizer=tokenizer, text_encoder=text_encoder, vae=vae, transformer=transformer, scheduler=scheduler`
`168`	`168`	`)`
`169`	`169`
`170`		`- self.vae_scale_factor = 2 ** (len(self.vae.config.encoder_width_list) - 1)`
	`170`	`+ self.vae_scale_factor = (`
	`171`	`+ 2 ** (len(self.vae.config.encoder_width_list) - 1) if hasattr(self, "vae") and self.vae is not None else 32`
	`172`	`+ )`
`171`	`173`	`self.image_processor = PixArtImageProcessor(vae_scale_factor=self.vae_scale_factor)`
`172`	`174`
`173`	`175`	`# Copied from diffusers.pipelines.pixart_alpha.pipeline_pixart_alpha.PixArtAlphaPipeline.encode_prompt with 120->300`