Huber loss not supported??

[Deathawaits4]

The readme states huber loss is both available for sd3.5 and Flux, but every time i try to train using it i get:

Huber schedule 'snr' is not supported with the current model.

[Ice-YY]

I believe only the exponential is supported for now

[Deathawaits4]

okay thanks for the answer, i tested and it seems to be the case. This is sad since snr yields better results than exponential. What i don't understand is the readme explicitly stating snr to be an option and it's also the default if not setting it, this is a little confusing.

[rockerBOO]

snr requires the noise scheduler to have alphas_cumprod but isn't set for FlowMatchEulerDiscreteScheduler. Beyond that I'm not sure what is involved. Maybe the README could be updated to be clearer.