ai-toolkit icon indicating copy to clipboard operation
ai-toolkit copied to clipboard

Flux lora training Error with Batch >1

Open Desm0nt opened this issue 1 year ago • 1 comments

On the current version in linux, if you set batch to more than 1, the start of Flux Lora training aborts with this error. The old version (commit https://github.com/ostris/ai-toolkit/commit/6d31c6db730a9cba3004eae1a3d7283c663d9295) does not have this problem.

Снимок

Desm0nt avatar Aug 14 '24 12:08 Desm0nt

this is not caused by the batch size, it's caused by linear_timesteps being true. Comment it out like in my example image

WarAnakin avatar Aug 14 '24 20:08 WarAnakin

This should be resolved with and without linear_timestep now.

jaretburkett avatar Aug 19 '24 03:08 jaretburkett