Add transformer config to model index (diffusers)
Browse files- model_index.json +5 -1
model_index.json
CHANGED
|
@@ -37,8 +37,12 @@
|
|
| 37 |
"transformers",
|
| 38 |
"PreTrainedTokenizerFast"
|
| 39 |
],
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
"vae": [
|
| 41 |
"diffusers",
|
| 42 |
"AutoencoderKL"
|
| 43 |
]
|
| 44 |
-
}
|
|
|
|
| 37 |
"transformers",
|
| 38 |
"PreTrainedTokenizerFast"
|
| 39 |
],
|
| 40 |
+
"transformer": [
|
| 41 |
+
"diffusers",
|
| 42 |
+
"HiDreamImageTransformer2DModel"
|
| 43 |
+
],
|
| 44 |
"vae": [
|
| 45 |
"diffusers",
|
| 46 |
"AutoencoderKL"
|
| 47 |
]
|
| 48 |
+
}
|