ge_size (int, optional, defaults to 512) —
The image size that the model was trained on. Use 512 for Stable Diffusion v1.X and Stable Diffusion v2
Base. Use 768 for Stable Diffusion v2.


prediction_type (str, optional) —
The prediction type that the model was trained on. Use 'epsilon' for Stable Diffusion v1.X and Stable
Diffusion v2 Base. Use 'v_prediction' for Stable Diffusion v2.


num_in_channels (int, optional, defaults to None) —
The num