Beinsezii commited on
Commit
09f6f6d
·
1 Parent(s): a865808

Add transformer config to model index (diffusers)

Browse files
Files changed (1) hide show
  1. 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
+ }