Update model_index.json
Browse files- model_index.json +2 -2
model_index.json
CHANGED
@@ -3,11 +3,11 @@
|
|
3 |
"_name_or_path": "ncsn-mnist",
|
4 |
"_diffusers_version": "0.32.0",
|
5 |
"scheduler": [
|
6 |
-
"
|
7 |
"AnnealedLangevinDynamicScheduler"
|
8 |
],
|
9 |
"unet": [
|
10 |
-
"
|
11 |
"UNet2DModelForNCSN"
|
12 |
]
|
13 |
}
|
|
|
3 |
"_name_or_path": "ncsn-mnist",
|
4 |
"_diffusers_version": "0.32.0",
|
5 |
"scheduler": [
|
6 |
+
"scheduling_ncsn",
|
7 |
"AnnealedLangevinDynamicScheduler"
|
8 |
],
|
9 |
"unet": [
|
10 |
+
"unet_2d_ncsn",
|
11 |
"UNet2DModelForNCSN"
|
12 |
]
|
13 |
}
|