bghira commited on
Commit
8ab7568
·
verified ·
1 Parent(s): 7833a5e

add transformer definition to model index

Browse files
Files changed (1) hide show
  1. model_index.json +4 -0
model_index.json CHANGED
@@ -1,6 +1,10 @@
1
  {
2
  "_class_name": "HiDreamImagePipeline",
3
  "_diffusers_version": "0.32.1",
 
 
 
 
4
  "scheduler": [
5
  "diffusers",
6
  "FlowMatchEulerDiscreteScheduler"
 
1
  {
2
  "_class_name": "HiDreamImagePipeline",
3
  "_diffusers_version": "0.32.1",
4
+ "transformer": [
5
+ "third_party.hidream",
6
+ "HiDreamImageTransformer2DModel"
7
+ ],
8
  "scheduler": [
9
  "diffusers",
10
  "FlowMatchEulerDiscreteScheduler"