Modal cloud training support, fixed typo in toolkit/scheduler.py, Schnell training support for Colab, issue #92 , issue #114

[martintomov]

• Added peft to requirements.txt since it’s needed for Schnell training. #92
• Fixed some missing parameters in the Colab notebook that were in the local config files. #114
• Threw in !nvidia-smi at the top of the notebooks to make sure the GPU is being used.
• Schnell training notebook for Colab
• Fixed typo in toolkit/scheduler.py - Commit: facor -> factor
• Training on Modal cloud + setup guide in README.md

Schnell training notebook Colab

Training on Modal

[ingl6rious]

where can i find this schnell training notebook?

[martintomov]

Until @jaretburkett merges, you can find it here: https://github.com/martintomov/ai-toolkit/blob/main/notebooks/FLUX_1_schnell_LoRA_Training.ipynb

[ingl6rious]

I didn't understand at first that I need A100 gpu, but I got it working and lora is good. Thanks martin

[jaretburkett]

Excellent work! Thank you so much! You are killing it with these PRs

[martintomov]

Happy to help the project

[frailloris]

@martintomov I got it running thanks to you! Appreciate it