devilismyfriend / StableTuner

Finetuning SD in style.
GNU Affero General Public License v3.0
670 stars 51 forks source link

Support training with different aspect ratios of than squares #82

Closed ProGamerGov closed 1 year ago

ProGamerGov commented 1 year ago

Currently it seems like you can only specific the size of the height and width dimensions together for a square shape. I would like to be able to independently specify different height and width values, for certain kinds of training data that for example uses a 1:2 ratio (768x1536) and requires that the entire image be visible to the model during each training step.

ProGamerGov commented 1 year ago

Actually, I think aspect ratio bucketing may already do this? https://github.com/NovelAI/novelai-aspect-ratio-bucketing

Jordach commented 1 year ago

StableTuner already supports aspect ratio bucketing.