Ngene787 commited on
Commit
9740cba
·
verified ·
1 Parent(s): b2a8baa

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
- "ClassConditionedUNet"
11
  ]
12
  }
 
6
  "DDPMScheduler"
7
  ],
8
  "unet": [
9
+ "diffusers",
10
+ "UNet2DConditionModel"
11
  ]
12
  }
unet/config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_class_name": "ClassConditionedUNet",
3
  "_diffusers_version": "0.32.2",
4
  "act_fn": "silu",
5
  "addition_embed_type": null,
@@ -70,4 +70,4 @@
70
  ],
71
  "upcast_attention": false,
72
  "use_linear_projection": false
73
- }
 
1
  {
2
+ "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.32.2",
4
  "act_fn": "silu",
5
  "addition_embed_type": null,
 
70
  ],
71
  "upcast_attention": false,
72
  "use_linear_projection": false
73
+ }