Stability-AI / StableCascade

Official Code for Stable Cascade
MIT License
6.44k stars 518 forks source link

Running LoRA in 24g vram but 30 seconds for single image #72

Open Suprhimp opened 4 months ago

Suprhimp commented 4 months ago

Hi, Thanks for the cool models first.

I just want to ask about the vram memory spend durring generation.

I tested with a10 gpu 24 gb vram

but when I checked the memory, it uses only 20 gb of vram image

and also the running time is quite slow for me should I run stage c, and stage b all the time?

image

In the example, generate 4 images spend only 26 seconds

스크린샷 2024-02-19 오전 11 50 12

Should I set some flag like --highvram or something? or what gpu should I use for the fast generate as example?

wodsoe commented 4 months ago

gdf sampler is very slow.

SimonAndMilky commented 4 months ago

how did you get it running in the first place? I was trying to train a Lora following the examples ( after setting the env variables for SLURMs all to 1, and fixed the - file error for webdataset), the code stuck at the progress bar after loading the models and images... I am using A100, and I can see around 18GiB of RAM is used with no 0% GPU usage.... are you following https://github.com/Stability-AI/StableCascade/blob/master/train/readme.md as well?

Suprhimp commented 4 months ago

I didn't start train yet, I just checked the output of the example :)

wen020 commented 4 months ago

@SimonAndMilky +1,Have you solved this problem?

wen020 commented 4 months ago

I didn't start train yet, I just checked the output of the example :)

Hi, I use code(https://github.com/Stability-AI/StableCascade/blob/master/inference/lora.ipynb) to generate images by lora, but the following error will be reported. image

I understand that this is caused by the model not being downloaded correctly. The code I used to download the model is as follows. Is there any problem with this operation?

download models :bash download_models.sh essential small-small bfloat16

Suprhimp commented 4 months ago

@wen020 Hi , I just run download models all, not the only essential