Update README.md

README.md

@@ -6,4 +6,38 @@ library_name: diffusers
 tags:
 - diffusion
 ---
-Generate some cute kitty cats!
+Generate some cute (cursed) kitty cats!
+
+import torch
+from diffusers import UNet2DModel, DDPMPipeline, DDPMScheduler
+import safetensors
+
+model_path = "ddpm-cats/model_epoch_9.safetensors"
+state_dict = safetensors.torch.load_file(model_path)
+image_size = 128
+
+model = UNet2DModel(
+    sample_size=image_size,
+    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.load_state_dict(state_dict)
+model.eval()
+
+scheduler = DDPMScheduler(
+    num_train_timesteps=1000,
+    beta_start=0.0001,
+    beta_end=0.02,
+    beta_schedule="linear",
+)
+
+pipeline = DDPMPipeline(
+    unet=model,
+    scheduler=scheduler,
+)
+
+image = pipeline(batch_size=1, generator=torch.manual_seed(0)).images[0].save("generated_image_128.png")