.   image_size (int, optional, defaults to 512) —
The image size the model was trained on. Use 512 for all Stable Diffusion v1 models and the Stable
Diffusion v2 base model. Use 768 for Stable Diffusion v2.   prediction_type (str, optional) —
The prediction type the model was trained on. Use 'epsilon' for all Stable Diffusion v1 models and
the Stable Diffusion v2 base model. Use 'v_prediction' for Stable Diffusion v2.   num_in_channels (int, op