Fix
Browse files- model_index.json +1 -1
model_index.json
CHANGED
@@ -24,7 +24,7 @@
|
|
24 |
],
|
25 |
"unet": [
|
26 |
"torch",
|
27 |
-
"
|
28 |
],
|
29 |
"vae": [
|
30 |
"diffusers",
|
|
|
24 |
],
|
25 |
"unet": [
|
26 |
"torch",
|
27 |
+
"UNet2DConditionModel"
|
28 |
],
|
29 |
"vae": [
|
30 |
"diffusers",
|