PromeAI

icon indicating a website link www.promeai.pro

icon company LibAI Lab icon location PromeAI has powerful AI-driven design assistants and an extensive controllable AIGC (C-AIGC) model style library, enabling you to easily create stunning AI art.

Results 6 comments of PromeAI

Flux Controlnet Train Example, will run out of memory on validation step

sorry respond late, i just test a few minutes earlier. tried with 10 validation images and didn't need more CUDA memory. So can you provide more detailed config? especially `resolution`,...

Flux Controlnet Train Example, will run out of memory on validation step

really confused me, ive tried install same transformer\accelerate versions of yours and works good. may i ask that what's your total gpu number? are you testing training on multi gpu?...

Flux Controlnet Train Example, will run out of memory on validation step

maybe it is about precision. I guess accelerate try convert params to bf16 but fail and remain fp32? it is a guess it be device-related issue, need to test on...

Flux Controlnet Train Example, will run out of memory on validation step

Consider using these two options `--use_adafactor ` `--gradient_checkpointing` @Night1099 @Trav1slaflame

Flux Controlnet Train Example, will run out of memory on validation step

> Consider using these two options `--use_adafactor ` `--gradient_checkpointing` @Night1099 @Trav1slaflame @universewill

Run FLUX-controlnet zero3 training failed: 'weight' must be 2-D

> Cc: [@PromeAIpro](https://github.com/PromeAIpro) using deepseek were experimental tries we had tested when writing this script, it was not included in code because it is not guaranteed work on any gpu...