Add Diffusers weights

feature_extractor/preprocessor_config.json

@@ -0,0 +1,44 @@
+{
+  "_valid_processor_keys": [
+    "images",
+    "do_resize",
+    "size",
+    "resample",
+    "do_center_crop",
+    "crop_size",
+    "do_rescale",
+    "rescale_factor",
+    "do_normalize",
+    "image_mean",
+    "image_std",
+    "do_convert_rgb",
+    "return_tensors",
+    "data_format",
+    "input_data_format"
+  ],
+  "crop_size": {
+    "height": 224,
+    "width": 224
+  },
+  "do_center_crop": true,
+  "do_convert_rgb": true,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.48145466,
+    0.4578275,
+    0.40821073
+  ],
+  "image_processor_type": "CLIPFeatureExtractor",
+  "image_std": [
+    0.26862954,
+    0.26130258,
+    0.27577711
+  ],
+  "resample": 3,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "shortest_edge": 224
+  }
+}

model_index.json

@@ -1,9 +1,9 @@
 {
   "_class_name": "StableDiffusionPipeline",
-  "_diffusers_version": "0.27.2",
+  "_diffusers_version": "0.28.0.dev0",
   "feature_extractor": [
-    null,
-    null
+    "transformers",
+    "CLIPFeatureExtractor"
   ],
   "image_encoder": [
     null,
@@ -11,12 +11,12 @@
   ],
   "requires_safety_checker": true,
   "safety_checker": [
-    null,
-    null
+    "stable_diffusion",
+    "StableDiffusionSafetyChecker"
   ],
   "scheduler": [
     "diffusers",
-    "DDIMScheduler"
+    "PNDMScheduler"
   ],
   "text_encoder": [
     "transformers",

safety_checker/config.json

@@ -0,0 +1,28 @@
+{
+  "_name_or_path": "CompVis/stable-diffusion-safety-checker",
+  "architectures": [
+    "StableDiffusionSafetyChecker"
+  ],
+  "initializer_factor": 1.0,
+  "logit_scale_init_value": 2.6592,
+  "model_type": "clip",
+  "projection_dim": 768,
+  "text_config": {
+    "dropout": 0.0,
+    "hidden_size": 768,
+    "intermediate_size": 3072,
+    "model_type": "clip_text_model",
+    "num_attention_heads": 12
+  },
+  "torch_dtype": "float32",
+  "transformers_version": "4.39.0.dev0",
+  "vision_config": {
+    "dropout": 0.0,
+    "hidden_size": 1024,
+    "intermediate_size": 4096,
+    "model_type": "clip_vision_model",
+    "num_attention_heads": 16,
+    "num_hidden_layers": 24,
+    "patch_size": 14
+  }
+}

safety_checker/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ec45b3c657b2756ae6392a8696cea33e4b31827916fc1ee2d23eb979738f5e43
+size 1216062778

scheduler/scheduler_config.json

@@ -1,21 +1,15 @@
 {
-  "_class_name": "DDIMScheduler",
-  "_diffusers_version": "0.27.2",
+  "_class_name": "PNDMScheduler",
+  "_diffusers_version": "0.28.0.dev0",
   "beta_end": 0.012,
   "beta_schedule": "scaled_linear",
   "beta_start": 0.00085,
   "clip_sample": false,
-  "clip_sample_range": 1.0,
-  "dynamic_thresholding_ratio": 0.995,
-  "interpolation_type": "linear",
   "num_train_timesteps": 1000,
   "prediction_type": "epsilon",
-  "rescale_betas_zero_snr": false,
-  "sample_max_value": 1.0,
   "set_alpha_to_one": false,
   "skip_prk_steps": true,
   "steps_offset": 1,
-  "thresholding": false,
   "timestep_spacing": "leading",
   "trained_betas": null
 }

text_encoder/config.json

@@ -19,6 +19,6 @@
   "pad_token_id": 1,
   "projection_dim": 768,
   "torch_dtype": "float32",
-  "transformers_version": "4.39.3",
+  "transformers_version": "4.39.0.dev0",
   "vocab_size": 49408
 }

text_encoder/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:056a9e82568bee336cb338076d242f71e242f3a737b1b5c12b9e578c76cfb6a4
+size 492306586

unet/config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "UNet2DConditionModel",
-  "_diffusers_version": "0.27.2",
+  "_diffusers_version": "0.28.0.dev0",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,

unet/diffusion_pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9c36a01e5f958aad17608d10cfdbc69490edfc00a3605e65fa8b333f600f4ae9
+size 3438366838

vae/config.json

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

vae/diffusion_pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:66d3d118c9806bd5dfc3da5c75f95e71dc57510fd7a3d34b7e8f510fcba3243e
+size 334712578