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

Remove override as the new simple loader works it out for us.

Browse Source
This commit is contained in:
gchapman
2024-06-23 19:44:32 +01:00
parent 7d0cb8fe31
commit 8f42946305
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
PixArt/loader.py
-2
View File
@@ -81,8 +81,6 @@ def load_pixart(model_path, model_conf=None):
device=model_management.get_torch_device() device=model_management.get_torch_device()
) )
model_conf.unet_config['depth'] = sum(key.endswith('mlp.fc1.weight') for key in state_dict.keys())
if model_conf.model_target == "PixArtMS": if model_conf.model_target == "PixArtMS":
from .models.PixArtMS import PixArtMS from .models.PixArtMS import PixArtMS
model.diffusion_model = PixArtMS(**model_conf.unet_config) model.diffusion_model = PixArtMS(**model_conf.unet_config)