Cannot access 'keep_in_vram'

[ginisksam]

Hi All,

Initial error was no module 'peft' found. Sorted that out with pip
then this error
UnboundLocalError: cannot access local variable 'keep_in_vram' where it is not associated with a value

Hope someone can guide me further.

Many thanks & Cheers

[xiaoyu9308]

I modified
ComfyUI/custom_nodes/ComfyUI-OmniGen/ailab_OmniGen.py


ComfyUI/custom_nodes/ComfyUI-OmniGen/OmniGen/scheduler.py

But when it comes to drawing, it will get stuck.

[quocvu]

Thanks @rarnu. Could you please send a PR

[MyUnclesWeirdThing]

I too am experiencing this issue ♥

[brianjking]

So does this fix things or not?

[goncami]

Same error after I installed manually diffusers, timm and peft dependencies...
Any fix?