alibaba-pai

/

Z-Image-Turbo-Fun-Controlnet-Union

like 267

Follow

Alibaba-PAI 1.28k

Here https://github.com/aigc-apps/VideoX-Fun/blob/6a95acc2b71b162d903104e2d1fc36936ec87fd0/scripts/z_image/train.py

Can I ask how much VRAM was required and how large the dataset was? I just need a ballpark number :)

We currently only provide the training code for Zimage itself. I’ll upload the training code for the control part soon.

Can I ask how much VRAM was required and how large the dataset was? I just need a ballpark number :)

8×A100 GPUs are sufficient to train the control model.

https://github.com/aigc-apps/VideoX-Fun/pull/388
Look here.

Excuse me, would it be possible to train controlnets on RTX 4090s possibly with some settings to lowering VRAM, offloading to RAM, indeed batch size like 1 etc or beefy 80 GB GPUs are needed?

And besides that, so Controlnet training script is not still available? Would it be? Really stoked to try to train custom one.

Thanks a lot for your great work and awesome models, you rock!