GitLab

This is to make it match the official checkpoint.

but faster.

The wrong dimensions were being checked, [1] and [2] are the image size.
not [2] and [3]. This results in an out-of-bounds error if one of them
actually matches.

load_model_gpu() is now used with the text encoder models instead of just
the unet.

This is faster on big text encoder models than running it on the CPU.