Add Pixart to AUTO_TEXT2IMAGE_PIPELINES_MAPPING (huggingface#5664)

Beinsezii · web-flow · commit fa1faf7f7922 · 2023-11-07T07:45:56.000+05:30
diff --git a/pipelines/auto_pipeline.py b/pipelines/auto_pipeline.py
@@ -43,6 +43,7 @@
     KandinskyV22Pipeline,
 )
 from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline
+from .pixart_alpha import PixArtAlphaPipeline
 from .stable_diffusion import (
     StableDiffusionImg2ImgPipeline,
     StableDiffusionInpaintPipeline,
@@ -67,6 +68,7 @@
         ("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetPipeline),
         ("wuerstchen", WuerstchenCombinedPipeline),
         ("lcm", LatentConsistencyModelPipeline),
+        ("pixart", PixArtAlphaPipeline),
     ]
 )
 

Original file line number	Diff line number	Diff line change
`@@ -43,6 +43,7 @@`
`43`	`43`	`KandinskyV22Pipeline,`
`44`	`44`	`)`
`45`	`45`	`from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline`
	`46`	`+from .pixart_alpha import PixArtAlphaPipeline`
`46`	`47`	`from .stable_diffusion import (`
`47`	`48`	`StableDiffusionImg2ImgPipeline,`
`48`	`49`	`StableDiffusionInpaintPipeline,`
`@@ -67,6 +68,7 @@`
`67`	`68`	`("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetPipeline),`
`68`	`69`	`("wuerstchen", WuerstchenCombinedPipeline),`
`69`	`70`	`("lcm", LatentConsistencyModelPipeline),`
	`71`	`+ ("pixart", PixArtAlphaPipeline),`
`70`	`72`	`]`
`71`	`73`	`)`
`72`	`74`