city96 / ComfyUI_ExtraModels

Support for miscellaneous image models. Currently supports: DiT, PixArt, HunYuanDiT, MiaoBi, and a few VAEs.
Apache License 2.0
362 stars 33 forks source link

Can't load pixart model in current version of comfy, pixart\loader.py throws error #80

Closed DanPli closed 1 month ago

DanPli commented 1 month ago

Comfy complains about the model patcher using an unexpected keyword argument current_device .

I fixed it for me by doing this:

In pixart\loader.py Line 114 find this: `

model_patcher = comfy.model_patcher.ModelPatcher(
    model,
    load_device = load_device,
    offload_device = offload_device,
    current_device = "cpu",
)

` remove

, current_device = "cpu",

So that it reads


    model_patcher = comfy.model_patcher.ModelPatcher(
        model,
        load_device = load_device,
        offload_device = offload_device
    )
city96 commented 1 month ago

Thanks, should be fixed.