End of training

model_index.json

@@ -1,10 +1,9 @@
 {
   "_class_name": "StableDiffusionPipeline",
   "_diffusers_version": "0.30.0.dev0",
-  "_name_or_path": "runwayml/stable-diffusion-v1-5",
   "feature_extractor": [
-    "transformers",
-    "CLIPImageProcessor"
+    null,
+    null
   ],
   "image_encoder": [
     null,
@@ -12,12 +11,12 @@
   ],
   "requires_safety_checker": true,
   "safety_checker": [
-    "stable_diffusion",
-    "StableDiffusionSafetyChecker"
+    null,
+    null
   ],
   "scheduler": [
     "diffusers",
-    "PNDMScheduler"
+    "DDPMScheduler"
   ],
   "text_encoder": [
     "transformers",
@@ -32,7 +31,7 @@
     "UNet2DConditionModel"
   ],
   "vae": [
-    "dsprites_components",
-    "DSpritesSingleChannelVAE"
+    "diffusers",
+    "AutoencoderKL"
   ]
 }

scheduler/scheduler_config.json

@@ -1,15 +1,21 @@
 {
-  "_class_name": "PNDMScheduler",
+  "_class_name": "DDPMScheduler",
   "_diffusers_version": "0.30.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,
   "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
+  "trained_betas": null,
+  "variance_type": "fixed_small"
 }

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c3639552864ef7a64b772474d1937ce3d9beef1984acb3fb26808f2446cdceec
+oid sha256:9bb311a195348942eab69ba25242a24942057b116e622c141b1fd7f9701fca80
 size 3438167536

vae/config.json

@@ -1,5 +1,5 @@
 {
-  "_class_name": "DSpritesSingleChannelVAE",
+  "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.30.0.dev0",
   "_name_or_path": "rcannizzaro/vae-dsprites",
   "act_fn": "silu",