search

kohya-ss / sd-scripts

Apache License 2.0
5.25k stars 871 forks source link

Flux how to prepare buckets latents? #1506

Open tristanwqy opened 2 months ago

tristanwqy commented 2 months ago

Do prepare buckets latents.py still work for flux-dev? If do so, what is the vae model path? models--black-forest-labs--FLUX.1-dev/vae/diffusion_pytorch_model.safetensors?

kohya-ss commented 2 months ago

Unfortunately prepare buckets latents.py doesn't support FLUX.1 yet.

tristanwqy commented 2 months ago

Unfortunately prepare buckets latents.py doesn't support FLUX.1 yet.

I think the prepare buckets latents is pretty important in multi gpu env, otherwise i have to cache it in the beginning of training, which means only one gpu doing their stuff, it's quite painful.

BenDes21 commented 2 months ago

Unfortunately prepare buckets latents.py doesn't support FLUX.1 yet.

I think the prepare buckets latents is pretty important in multi gpu env, otherwise i have to cache it in the beginning of training, which means only one gpu doing their stuff, it's quite painful.

any news ? still not supported ?