Add model index config for transformer to make format uniform

#11
by a-r-r-o-w HF Staff - opened
Files changed (1) hide show
  1. 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"