nodes/city96-ComfyUI_ExtraModels
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

PixArt sigma

Browse Source 
Mentioned in #19
Also removes old example sampler since updating it would be too much overhead.
Pytorch attention is still wonky.
This commit is contained in:
City
2024-04-07 17:55:33 +02:00
parent 059c87ebb5
commit c325dec0d3
11 changed files with 266 additions and 2745 deletions
Download Patch File Download Diff File Expand all files Collapse all files
PixArt/loader.py
+4
View File
@@ -82,6 +82,10 @@ def load_pixart(model_path, model_conf):
elif model_conf.model_target == "PixArt":
from .models.PixArt import PixArt
model.diffusion_model = PixArt(**model_conf.unet_config)
elif model_conf.model_target == "PixArtMSSigma":
from .models.PixArtMS import PixArtMS
model.diffusion_model = PixArtMS(**model_conf.unet_config)
model.latent_format = comfy.latent_formats.SDXL()
elif model_conf.model_target == "ControlPixArtMSHalf":
from .models.PixArtMS import PixArtMS
from .models.pixart_controlnet import ControlPixArtMSHalf