Add Diffusers weights (#3)

- Add Diffusers weights (f0e8974df8144f06377349e4b99be4297d17c9ef)

scheduler/scheduler_config.json

@@ -6,7 +6,7 @@
   "beta_start": 0.00085,
   "clip_sample": false,
   "num_train_timesteps": 1000,
-  "prediction_type": "v_prediction",
+  "prediction_type": "epsilon",
   "set_alpha_to_one": false,
   "skip_prk_steps": true,
   "steps_offset": 1,

unet/config.json

@@ -34,13 +34,13 @@
   "num_class_embeds": null,
   "only_cross_attention": false,
   "out_channels": 4,
-  "sample_size": 96,
+  "sample_size": 64,
   "up_block_types": [
     "UpBlock2D",
     "CrossAttnUpBlock2D",
     "CrossAttnUpBlock2D",
     "CrossAttnUpBlock2D"
   ],
-  "upcast_attention": true,
+  "upcast_attention": false,
   "use_linear_projection": true
 }

vae/config.json

@@ -19,7 +19,7 @@
   "layers_per_block": 2,
   "norm_num_groups": 32,
   "out_channels": 3,
-  "sample_size": 768,
+  "sample_size": 512,
   "up_block_types": [
     "UpDecoderBlock2D",
     "UpDecoderBlock2D",