Upload 4 files

model_index.json

@@ -1,9 +1,9 @@
 {
-  "_class_name": "Sketch2ImgPipeline",
+  "_class_name": "DDPMSketch2ImgPipeline",
   "_diffusers_version": "0.11.1",
   "scheduler": [
     "diffusers",
-    "DDIMScheduler"
+    "DDPMScheduler"
   ],
   "unet": [
     "diffusers",

scheduler/scheduler_config.json

@@ -1,5 +1,5 @@
 {
-  "_class_name": "DDIMScheduler",
+  "_class_name": "DDPMScheduler",
   "_diffusers_version": "0.11.1",
   "beta_end": 0.02,
   "beta_schedule": "linear",
@@ -7,7 +7,6 @@
   "clip_sample": true,
   "num_train_timesteps": 1000,
   "prediction_type": "epsilon",
-  "set_alpha_to_one": true,
-  "steps_offset": 0,
-  "trained_betas": null
+  "trained_betas": null,
+  "variance_type": "fixed_small"
 }

unet/config.json

@@ -1,6 +1,7 @@
 {
   "_class_name": "UNet2DModel",
   "_diffusers_version": "0.11.1",
+  "_name_or_path": "model/init_s2i_fmnist_5epochs\\unet",
   "act_fn": "silu",
   "add_attention": true,
   "attention_head_dim": 8,
@@ -17,7 +18,7 @@
   "flip_sin_to_cos": true,
   "freq_shift": 0,
   "in_channels": 2,
-  "layers_per_block": 1,
+  "layers_per_block": 2,
   "mid_block_scale_factor": 1,
   "norm_eps": 1e-05,
   "norm_num_groups": 32,

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c2ebed4b225abf91e5ca8a7bfcb9cf02d8d9c6ce7df19d0b3144e2e7ca2d0797
-size 2861505
+oid sha256:cdab880696cbe5e9867543092868e5c5f924a14a3f5c4e6fd6518957c1c87e3b
+size 4087425