Add Diffusers weights (#3)

- Add Diffusers weights (568a6d679ca2f347e78f2e80a5abe18c442531da)


Co-authored-by: Alex Poulos <[email protected]>

feature_extractor/preprocessor_config.json

@@ -1,8 +1,12 @@
 {
-  "crop_size": 224,
+  "crop_size": {
+    "height": 224,
+    "width": 224
+  },
   "do_center_crop": true,
   "do_convert_rgb": true,
   "do_normalize": true,
+  "do_rescale": true,
   "do_resize": true,
   "feature_extractor_type": "CLIPFeatureExtractor",
   "image_mean": [
@@ -10,11 +14,15 @@
     0.4578275,
     0.40821073
   ],
+  "image_processor_type": "CLIPImageProcessor",
   "image_std": [
     0.26862954,
     0.26130258,
     0.27577711
   ],
   "resample": 3,
-  "size": 224
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "shortest_edge": 224
+  }
 }

model_index.json

@@ -1,10 +1,11 @@
 {
   "_class_name": "StableDiffusionPipeline",
-  "_diffusers_version": "0.8.0.dev0",
+  "_diffusers_version": "0.9.0",
   "feature_extractor": [
     "transformers",
-    "CLIPFeatureExtractor"
+    "CLIPImageProcessor"
   ],
+  "requires_safety_checker": true,
   "safety_checker": [
     "stable_diffusion",
     "StableDiffusionSafetyChecker"

safety_checker/config.json

@@ -1,5 +1,5 @@
 {
-  "_commit_hash": "4bb648a606ef040e7685bde262611766a5fdd67b",
+  "_commit_hash": "cb41f3a270d63d454d385fc2e4f571c487c253c5",
   "_name_or_path": "CompVis/stable-diffusion-safety-checker",
   "architectures": [
     "StableDiffusionSafetyChecker"
@@ -62,6 +62,7 @@
     "pad_token_id": 1,
     "prefix": null,
     "problem_type": null,
+    "projection_dim": 512,
     "pruned_heads": {},
     "remove_invalid_values": false,
     "repetition_penalty": 1.0,
@@ -79,7 +80,7 @@
     "top_p": 1.0,
     "torch_dtype": null,
     "torchscript": false,
-    "transformers_version": "4.24.0",
+    "transformers_version": "4.25.0.dev0",
     "typical_p": 1.0,
     "use_bfloat16": false,
     "vocab_size": 49408
@@ -148,6 +149,7 @@
     "patch_size": 14,
     "prefix": null,
     "problem_type": null,
+    "projection_dim": 512,
     "pruned_heads": {},
     "remove_invalid_values": false,
     "repetition_penalty": 1.0,
@@ -165,7 +167,7 @@
     "top_p": 1.0,
     "torch_dtype": null,
     "torchscript": false,
-    "transformers_version": "4.24.0",
+    "transformers_version": "4.25.0.dev0",
     "typical_p": 1.0,
     "use_bfloat16": false
   },

safety_checker/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:16d28f2b37109f222cdc33620fdd262102ac32112be0352a7f77e9614b35a394
-size 1216064769
+oid sha256:193490b58ef62739077262e833bf091c66c29488058681ac25cf7df3d8190974
+size 1216061799

scheduler/scheduler_config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "PNDMScheduler",
-  "_diffusers_version": "0.8.0.dev0",
+  "_diffusers_version": "0.9.0",
   "beta_end": 0.012,
   "beta_schedule": "scaled_linear",
   "beta_start": 0.00085,

text_encoder/config.json

@@ -20,6 +20,6 @@
   "pad_token_id": 1,
   "projection_dim": 768,
   "torch_dtype": "float32",
-  "transformers_version": "4.24.0",
+  "transformers_version": "4.25.0.dev0",
   "vocab_size": 49408
 }

text_encoder/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dc7aab76789ec521129f3e04a1ed6cac24d61b050e9a71356dce6cd58d1b9e01
-size 492307041
+oid sha256:f3f4a24ea86b32e2ab91c035bd3377777bc71dc71102906625e3266025150580
+size 492305335

unet/config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "UNet2DConditionModel",
-  "_diffusers_version": "0.8.0.dev0",
+  "_diffusers_version": "0.9.0",
   "act_fn": "silu",
   "attention_head_dim": 8,
   "block_out_channels": [
@@ -18,6 +18,7 @@
     "DownBlock2D"
   ],
   "downsample_padding": 1,
+  "dual_cross_attention": false,
   "flip_sin_to_cos": true,
   "freq_shift": 0,
   "in_channels": 4,
@@ -25,12 +26,15 @@
   "mid_block_scale_factor": 1,
   "norm_eps": 1e-05,
   "norm_num_groups": 32,
+  "num_class_embeds": null,
+  "only_cross_attention": false,
   "out_channels": 4,
-  "sample_size": 32,
+  "sample_size": 64,
   "up_block_types": [
     "UpBlock2D",
     "CrossAttnUpBlock2D",
     "CrossAttnUpBlock2D",
     "CrossAttnUpBlock2D"
-  ]
+  ],
+  "use_linear_projection": false
 }

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f9b1df8e3f2a61392024f3e6581fd7f71d54c3aba4735368f9ad4e52c0b3bc9f
-size 3438366373
+oid sha256:0e3bcefa09442056f08552ff710d9c1e8742ee75a73921ee904fbfa1357a91a2
+size 3438354725

vae/config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "AutoencoderKL",
-  "_diffusers_version": "0.8.0.dev0",
+  "_diffusers_version": "0.9.0",
   "act_fn": "silu",
   "block_out_channels": [
     128,

vae/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dcf4507d99b88db73f3916e2a20169fe74ada6b5582e9af56cfa80f5f3141765
-size 334711857
+oid sha256:3a407758a137a1c6bfcd11e154eb06e4dcffa996009e46eb27d1af04a81e7e74
+size 334707217