Update README.md

README.md

@@ -21,8 +21,8 @@ class MyUNet2DModel(UNet2DModel):
             up_block_types=("AttnUpBlock2D", "AttnUpBlock2D", "AttnUpBlock2D", "UpBlock2D"),
             act_fn="silu",
             block_out_channels=block_out_channels,
-            layers_per_block=2,  # default=2
-            norm_num_groups=32,  # default=32
+            layers_per_block=2, 
+            norm_num_groups=32, 
             norm_eps=1e-6,
         )
         self.time_proj = GaussianFourierProjection(embedding_size=block_out_channels[0], scale=16,