Ngene787 commited on
Commit
db82999
·
verified ·
1 Parent(s): f31eed7

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. model_index.json +2 -2
  2. unet/config.json +2 -2
model_index.json CHANGED
@@ -6,7 +6,7 @@
6
  "DDPMScheduler"
7
  ],
8
  "unet": [
9
- "models/unet",
10
- "ADMUNet"
11
  ]
12
  }
 
6
  "DDPMScheduler"
7
  ],
8
  "unet": [
9
+ "diffusers",
10
+ "UNet2DModel"
11
  ]
12
  }
unet/config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_class_name": "ADMUNet",
3
  "_diffusers_version": "0.32.2",
4
  "act_fn": "silu",
5
  "add_attention": true,
@@ -49,4 +49,4 @@
49
  "UpBlock2D"
50
  ],
51
  "upsample_type": "resnet"
52
- }
 
1
  {
2
+ "_class_name": "UNet2DModel",
3
  "_diffusers_version": "0.32.2",
4
  "act_fn": "silu",
5
  "add_attention": true,
 
49
  "UpBlock2D"
50
  ],
51
  "upsample_type": "resnet"
52
+ }