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 |
+
}
|