mskrt commited on
Commit
45fe31c
verified
1 Parent(s): 4eb194a

Update model_index.json

Browse files
Files changed (1) hide show
  1. model_index.json +5 -5
model_index.json CHANGED
@@ -3,10 +3,10 @@
3
  "_diffusers_version": "0.31.0",
4
  "pipeline_tag": "custom",
5
  "model_config": {
6
- "unet": ["diffusers", "UNet2DConditionModel"],
7
- "vae": ["diffusers", "AutoencoderKL"],
8
- "scheduler": ["diffusers", "EulerDiscreteScheduler"],
9
- "tokenizer": ["transformers", "CLIPTokenizer"],
10
- "text_encoder": ["transformers", "CLIPTextModel"]
11
  }
12
  }
 
3
  "_diffusers_version": "0.31.0",
4
  "pipeline_tag": "custom",
5
  "model_config": {
6
+ "unet": "unet",
7
+ "vae": "vae",
8
+ "scheduler": "scheduler",
9
+ "tokenizer": "tokenizer",
10
+ "text_encoder": "text_encoder"
11
  }
12
  }