Dinghuai
/

flow-matching-cifar10

Model card Files Files and versions Community

Dinghuai commited on Jul 10

Commit

4ab4441

•

1 Parent(s): c677162

Update README.md

Files changed (1) hide show

README.md +0 -1

README.md CHANGED Viewed

@@ -13,7 +13,6 @@ from diffusers.models.embeddings import GaussianFourierProjection
 class MyUNet2DModel(UNet2DModel):
     def __init__(self, *args, **kwargs):
-        # super(MyUNet2DModel, self).__init__(*args, **kwargs)
         block_out_channels = (128, 256, 256, 256)
         super(MyUNet2DModel, self).__init__(
             in_channels=3, out_channels=3,

 class MyUNet2DModel(UNet2DModel):
     def __init__(self, *args, **kwargs):
         block_out_channels = (128, 256, 256, 256)
         super(MyUNet2DModel, self).__init__(
             in_channels=3, out_channels=3,