Hugging Face's logo

Dinghuai
/
flow-matching-cifar10

Diffusers
Safetensors
diffusion
flow-matching
Model card Files Community
Dinghuai commited on
Commit
4ab4441
·
verified ·
1 Parent(s): c677162

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -1
README.md CHANGED
@@ -13,7 +13,6 @@ from diffusers.models.embeddings import GaussianFourierProjection
13
 
14
  class MyUNet2DModel(UNet2DModel):
15
  def __init__(self, *args, **kwargs):
16
- # super(MyUNet2DModel, self).__init__(*args, **kwargs)
17
  block_out_channels = (128, 256, 256, 256)
18
  super(MyUNet2DModel, self).__init__(
19
  in_channels=3, out_channels=3,
 
13
 
14
  class MyUNet2DModel(UNet2DModel):
15
  def __init__(self, *args, **kwargs):
 
16
  block_out_channels = (128, 256, 256, 256)
17
  super(MyUNet2DModel, self).__init__(
18
  in_channels=3, out_channels=3,