search

AmericanPresidentJimmyCarter / simple-flux-lora-training

MIT License
62 stars 6 forks source link

Torch.tensor issue #2

Closed enterv0id closed 2 months ago

enterv0id commented 2 months ago

Thanks for your work, it helps a lot! ❤ My first lora came out 10/10 yesterday, following your guide.

However, today I'm getting these warnings on each training steps:

Steps:   1%|_                                  | 64/5000 [19:45<24:36:41, 17.95s/it, lr=0.0001, step_loss=0.384]Passing `txt_ids` 3d torch.Tensor is deprecated.Please remove the batch dimension and pass it as a 2d torch Tensor
Passing `img_ids` 3d torch.Tensor is deprecated.Please remove the batch dimension and pass it as a 2d torch Tensor

Is there any fix for that? Also, does it affects the training itself?

AmericanPresidentJimmyCarter commented 2 months ago

It does not affect training. It's just a deprecation warning in SimpleTuner that will be patched in the next release. You can ignore it.

enterv0id commented 2 months ago

Thank you!