Add PixArtSigmaPipeline to AutoPipeline mapping (#7783)

Beinsezii · web-flow · commit 0d2d424fbef9 · 2024-04-26T09:10:20.000-10:00
diff --git a/src/diffusers/pipelines/auto_pipeline.py b/src/diffusers/pipelines/auto_pipeline.py
@@ -45,7 +45,7 @@
 )
 from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline
 from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline
-from .pixart_alpha import PixArtAlphaPipeline
+from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline
 from .stable_cascade import StableCascadeCombinedPipeline, StableCascadeDecoderPipeline
 from .stable_diffusion import (
     StableDiffusionImg2ImgPipeline,
@@ -73,7 +73,8 @@
         ("wuerstchen", WuerstchenCombinedPipeline),
         ("cascade", StableCascadeCombinedPipeline),
         ("lcm", LatentConsistencyModelPipeline),
-        ("pixart", PixArtAlphaPipeline),
+        ("pixart-alpha", PixArtAlphaPipeline),
+        ("pixart-sigma", PixArtSigmaPipeline),
     ]
 )
 

Original file line number	Diff line number	Diff line change
`@@ -45,7 +45,7 @@`
`45`	`45`	`)`
`46`	`46`	`from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline`
`47`	`47`	`from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline`
`48`		`-from .pixart_alpha import PixArtAlphaPipeline`
	`48`	`+from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline`
`49`	`49`	`from .stable_cascade import StableCascadeCombinedPipeline, StableCascadeDecoderPipeline`
`50`	`50`	`from .stable_diffusion import (`
`51`	`51`	`StableDiffusionImg2ImgPipeline,`
`@@ -73,7 +73,8 @@`
`73`	`73`	`("wuerstchen", WuerstchenCombinedPipeline),`
`74`	`74`	`("cascade", StableCascadeCombinedPipeline),`
`75`	`75`	`("lcm", LatentConsistencyModelPipeline),`
`76`		`- ("pixart", PixArtAlphaPipeline),`
	`76`	`+ ("pixart-alpha", PixArtAlphaPipeline),`
	`77`	`+ ("pixart-sigma", PixArtSigmaPipeline),`
`77`	`78`	`]`
`78`	`79`	`)`
`79`	`80`