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