Update config.json

config.json

@@ -1,31 +1,19 @@
 {
-  "_class_name": "DDPMPipeline",
-  "model_name": "DDPMCatsPipeline",
-  "pipeline_type": "diffusers.DDPMPipeline",
-  "description": "A UNet2DModel trained on the 'cat' dataset using the DDPM algorithm for image generation.",
-  "model_weights": {
-    "type": "safetensors",
-    "path": "model_epoch_9.safetensors"
-  },
-  "scheduler": [
-    "diffusers",
-    "DDPMScheduler"
-  ],
-  "unet": [
-    "diffusers",
-    "UNet2DModel"
-  ],
-  "num_train_timesteps": 1000,
-  "beta_start": 0.0001,
-  "beta_end": 0.02,
-  "beta_schedule": "linear",
-  "unet_config": {
-    "sample_size": 128,
-    "in_channels": 3,
-    "out_channels": 3,
-    "layers_per_block": 2,
-    "block_out_channels": [128, 128, 256, 256, 512, 512],
-    "down_block_types": ["DownBlock2D", "DownBlock2D", "DownBlock2D", "DownBlock2D", "AttnDownBlock2D", "DownBlock2D"],
-    "up_block_types": ["UpBlock2D", "AttnUpBlock2D", "UpBlock2D", "UpBlock2D", "UpBlock2D", "UpBlock2D"]
-  }
+    "model_path": "ddpm-cats/model_epoch_9.safetensors",
+    "image_size": 128,
+    "model": {
+        "sample_size": 128,
+        "in_channels": 3,
+        "out_channels": 3,
+        "layers_per_block": 2,
+        "block_out_channels": [128, 128, 256, 256, 512, 512],
+        "down_block_types": ["DownBlock2D", "DownBlock2D", "DownBlock2D", "DownBlock2D", "AttnDownBlock2D", "DownBlock2D"],
+        "up_block_types": ["UpBlock2D", "AttnUpBlock2D", "UpBlock2D", "UpBlock2D", "UpBlock2D", "UpBlock2D"]
+    },
+    "scheduler": {
+        "num_train_timesteps": 1000,
+        "beta_start": 0.0001,
+        "beta_end": 0.02,
+        "beta_schedule": "linear"
+    }
 }