Add model index config for transformer to make format uniform (#11)
Browse files- Add model index config for transformer to make format uniform (0fdb8e6832e1a74b76fbc1920cbe83e3898169e1)
Co-authored-by: Aryan V S <[email protected]>
- model_index.json +4 -0
model_index.json
CHANGED
@@ -37,6 +37,10 @@
|
|
37 |
"transformers",
|
38 |
"PreTrainedTokenizerFast"
|
39 |
],
|
|
|
|
|
|
|
|
|
40 |
"vae": [
|
41 |
"diffusers",
|
42 |
"AutoencoderKL"
|
|
|
37 |
"transformers",
|
38 |
"PreTrainedTokenizerFast"
|
39 |
],
|
40 |
+
"transformer": [
|
41 |
+
"diffusers",
|
42 |
+
"HiDreamImageTransformer2DModel"
|
43 |
+
],
|
44 |
"vae": [
|
45 |
"diffusers",
|
46 |
"AutoencoderKL"
|