NOT ABLE TO LORA FINE TUNE ON SANA SPRINT MODELS

[pgagi-abhishek-khatri]

I want to LoRA fine-tune on the Efficient-Large-Model/Sana_Sprint_1.6B_1024px_diffusers, but I guess currently we can LoRA fine-tune on only the Efficient-Large-Model/Sana_1600M_1024px_BF16. I found the script here: https://github.com/NVlabs/Sana/blob/main/asset/docs/sana_lora_dreambooth.md

But when I replace the model with the sprint model, it produces some errors like invalid parameter passes 'batch size'. This code is not yet meant for sprint models. But then, is it impossible to have LoRA DreamBooth Fine Tuning on Sana Sprint Models?

Or am I doing something wrong?