End of training

model_index.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "StableDiffusionPipeline",
   "_diffusers_version": "0.31.0",
-  "_name_or_path": "CompVis/stable-diffusion-v1-4",
+  "_name_or_path": "runwayml/stable-diffusion-inpainting",
   "feature_extractor": [
     "transformers",
     "CLIPImageProcessor"
@@ -17,7 +17,7 @@
   ],
   "scheduler": [
     "diffusers",
-    "PNDMScheduler"
+    "DDIMScheduler"
   ],
   "text_encoder": [
     "transformers",

safety_checker/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "C:\\Users\\bsmolka\\.cache\\huggingface\\hub\\models--CompVis--stable-diffusion-v1-4\\snapshots\\133a221b8aa7292a167afc5127cb63fb5005638b\\safety_checker",
+  "_name_or_path": "C:\\Users\\bsmolka\\.cache\\huggingface\\hub\\models--runwayml--stable-diffusion-inpainting\\snapshots\\8a4288a76071f7280aedbdb3253bdb9e9d5d84bb\\safety_checker",
   "architectures": [
     "StableDiffusionSafetyChecker"
   ],

scheduler/scheduler_config.json

@@ -1,15 +1,20 @@
 {
-  "_class_name": "PNDMScheduler",
+  "_class_name": "DDIMScheduler",
   "_diffusers_version": "0.31.0",
   "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
 }

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "CompVis/stable-diffusion-v1-4",
+  "_name_or_path": "runwayml/stable-diffusion-inpainting",
   "architectures": [
     "CLIPTextModel"
   ],
@@ -18,7 +18,7 @@
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
   "pad_token_id": 1,
-  "projection_dim": 512,
+  "projection_dim": 768,
   "torch_dtype": "float32",
   "transformers_version": "4.46.2",
   "vocab_size": 49408

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.31.0",
-  "_name_or_path": "CompVis/stable-diffusion-v1-4",
+  "_name_or_path": "runwayml/stable-diffusion-inpainting",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,
@@ -34,7 +34,7 @@
   "encoder_hid_dim_type": null,
   "flip_sin_to_cos": true,
   "freq_shift": 0,
-  "in_channels": 4,
+  "in_channels": 9,
   "layers_per_block": 2,
   "mid_block_only_cross_attention": null,
   "mid_block_scale_factor": 1,

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f8ea219bace3dde1471fc52ca1208e54e41f686bf78784240438a29ae96690cc
-size 3438167536
+oid sha256:f3e2e7da863390677c6e5b8b6b96f338fa3b1f6b1a551d6b8b2d997e847ad35d
+size 3438225152

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.31.0",
-  "_name_or_path": "C:\\Users\\bsmolka\\.cache\\huggingface\\hub\\models--CompVis--stable-diffusion-v1-4\\snapshots\\133a221b8aa7292a167afc5127cb63fb5005638b\\vae",
+  "_name_or_path": "C:\\Users\\bsmolka\\.cache\\huggingface\\hub\\models--runwayml--stable-diffusion-inpainting\\snapshots\\8a4288a76071f7280aedbdb3253bdb9e9d5d84bb\\vae",
   "act_fn": "silu",
   "block_out_channels": [
     128,
@@ -24,7 +24,7 @@
   "mid_block_add_attention": true,
   "norm_num_groups": 32,
   "out_channels": 3,
-  "sample_size": 512,
+  "sample_size": 256,
   "scaling_factor": 0.18215,
   "shift_factor": null,
   "up_block_types": [