Add dimention for trained models

Some models (checkpoints, LoRA) are trained on 512x512, some are 512x768, some are even 2048x2048

If lower trained models are used for generation of large image (without upscale) they tend to make repetitions.

So adding a training dimension data to models would help resolve this problem.