invoke-ai / invoke-training

https://invoke-ai.github.io/invoke-training/
Apache License 2.0
120 stars 14 forks source link

initial addition of Finetune tab #127

Closed hipsterusername closed 6 months ago

hipsterusername commented 6 months ago

Initial stab at Finetune tab for SDXL.

RyanJDick commented 6 months ago

I did a quick smoke test of this. The only thing I noticed is that the use_8bit config isn't supported in the UI yet. This is necessary in order for SDXL finetuning to fit in 24GB of VRAM.

I can add that and get this merged.

Out of curiosity, was it working as-is for you on Windows? If yes, it's probably doing some VRAM swapping.