search

KohakuBlueleaf / a1111-sd-webui-locon

A extension for loading LyCORIS model in sd-webui
Apache License 2.0
502 stars 110 forks source link

VRAM leak when extension is installed #13

Open SidAbhi opened 1 year ago

SidAbhi commented 1 year ago

When this extension is installed, after a few times generating images with either lora or locon the vram will be filled and there's no way to fix without restarting. Uninstalling the extension fixes the problem.

COMMANDLINE_ARGS= --listen --medvram --xformers --api

lozn00 commented 1 year ago

oh

KohakuBlueleaf commented 1 year ago

you mean also the normal lora?

SidAbhi commented 1 year ago

Yes even just using only regular loras. I thought at first maybe the medvram/xformers settings didn't know how to flush locon/lycoris, but even just using regular loras the VRAM would eventually fill and won't empty. However I have not tried not using any loras at all.

Also it's somewhat random when this happens. Sometimes I could generate 100 images before it breaks, sometimes it happens on the first couple of images.

oh and the gpu is 8GB 3060TI and here's the extension list I use, maybe it could also be a conflict with something here (I did make sure to not use any of the other extensions when testing it)

posex Txt2Vectorgraphics sd-webui-depth-lib sd-dynamic-prompts stable-diffusion-webui-composable-lora sd-extension-steps-animation ABG_extension embedding-inspector sd-webui-ar stable-diffusion-webui-aesthetic-image-scorer stable-diffusion-webui-two-shot-old sd-webui-additional-networks sd-webui-lora-block-weight stable-diffusion-webui-images-browser Stable-Diffusion-Webui-Civitai-Helper sd-webui-cutoff Auto-Photoshop-StableDiffusion-Plugin openOutpaint sd-dynamic-thresholding stable-diffusion-webui-Prompt_Generator canvas-zoom stable-diffusion-webui-two-shot sd-webui-controlnet openpose-editor a1111-stable-diffusion-webui-vram-estimator stable-diffusion-webui-cafe-aesthetic

SidAbhi commented 1 year ago

Tried this extension again and I am no longer having this issue with the latest version of webui, the latest version of this extension, pytorch 2, and xformers 0.18, and updated all my extensions. I don't really know which of those fixed it, but it is fixed for me.