search

BlenderNeko / ComfyUI_Noise

6 nodes for ComfyUI that allows for more control and flexibility over noise to do e.g. variations or "un-sampling"
GNU General Public License v3.0
222 stars 19 forks source link

Unsampler broken by comfyui update #11

Closed antiquechrono closed 8 months ago

antiquechrono commented 8 months ago

Unsampler was broken in the following comfyui commit by the removal of the batch_area_memory method https://github.com/comfyanonymous/ComfyUI/commit/dd4ba68b6e93a562d9499eff34e50dbbbc8714e7

The removed method is called here https://github.com/BlenderNeko/ComfyUI_Noise/blob/a8c99723fea7c3c6af377cd9443a931482b671d6/nodes.py#L225

Switching out the old call for the new one seems to work for me

comfy.model_management.load_models_gpu([model] + models, model.memory_required(noise.shape) + inference_memory)