Add support for dataautogpt3/PixArt-Sigma-900M to PixArt-Alpha module

Woolverine94 · Jul 15, 2024 · 676c2dd · 676c2dd
1 parent d159876
commit 676c2dd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/ressources/txt2img_paa.py b/ressources/txt2img_paa.py
@@ -27,6 +27,7 @@
     "PixArt-alpha/PixArt-XL-2-512x512",
     "PixArt-alpha/PixArt-XL-2-1024-MS",
     "PixArt-alpha/PixArt-Sigma-XL-2-1024-MS",
+    "dataautogpt3/PixArt-Sigma-900M",
     "PixArt-alpha/PixArt-LCM-XL-2-1024-MS",
     "Luo-Yihong/yoso_pixart512",
     "Luo-Yihong/yoso_pixart1024",
@@ -168,7 +169,7 @@ def image_txt2img_paa(
         pipe_txt2img_paa.scheduler = LCMScheduler.from_config(pipe_txt2img_paa.scheduler.config)
         pipe_txt2img_paa.scheduler.config.timestep_spacing = "trailing"
 
-    elif ("PIXART-SIGMA-XL-2" in modelid_txt2img_paa.upper()):
+    elif ("PIXART-SIGMA" in modelid_txt2img_paa.upper()):
         if modelid_txt2img_paa[0:9] == "./models/" :
             pipe_txt2img_paa = PixArtSigmaPipeline.from_single_file(
                 modelid_txt2img_paa,

diff --git a/webui.py b/webui.py
@@ -581,7 +581,7 @@ def change_model_type_txt2img_paa(model_txt2img_paa):
         return sampler_txt2img_paa.update(value="LCM", interactive=False), width_txt2img_paa.update(value=biniou_global_sdxl_width), height_txt2img_paa.update(value=biniou_global_sdxl_height), guidance_scale_txt2img_paa.update(value=0.0), num_inference_step_txt2img_paa.update(value=4)
     elif model_txt2img_paa == "PixArt-alpha/PixArt-LCM-XL-2-1024-MS":
         return sampler_txt2img_paa.update(value="LCM", interactive=False), width_txt2img_paa.update(value=biniou_global_sdxl_width), height_txt2img_paa.update(value=biniou_global_sdxl_height), guidance_scale_txt2img_paa.update(value=0.0), num_inference_step_txt2img_paa.update(value=4)
-    elif model_txt2img_paa == "PixArt-alpha/PixArt-Sigma-XL-2-1024-MS":
+    elif model_txt2img_paa == "PixArt-alpha/PixArt-Sigma-XL-2-1024-MS" or model_txt2img_paa == "dataautogpt3/PixArt-Sigma-900M":
         return sampler_txt2img_paa.update(value="UniPC", interactive=True), width_txt2img_paa.update(value=biniou_global_sdxl_width), height_txt2img_paa.update(value=biniou_global_sdxl_height), guidance_scale_txt2img_paa.update(value=7.0), num_inference_step_txt2img_paa.update(value=15)
     elif model_txt2img_paa == "PixArt-alpha/PixArt-Sigma-XL-2-2K-MS":
         return sampler_txt2img_paa.update(value="UniPC", interactive=True), width_txt2img_paa.update(value=2048), height_txt2img_paa.update(value=2048), guidance_scale_txt2img_paa.update(value=7.0), num_inference_step_txt2img_paa.update(value=15)