mskrt commited on
Commit
094c541
verified
1 Parent(s): 24b2ab0

Update model_index.json

Browse files
Files changed (1) hide show
  1. model_index.json +8 -11
model_index.json CHANGED
@@ -1,15 +1,12 @@
1
  {
2
  "_class_name": "SuperDiffPipeline",
3
  "_diffusers_version": "0.31.0",
4
- "batch_size": null,
5
- "device": "cuda",
6
- "guidance_scale": null,
7
- "lift": null,
8
- "num_inference_steps": null,
9
- "scheduler": "EulerDiscreteScheduler",
10
- "seed": null,
11
- "text_encoder": "CLIPTextModel",
12
- "tokenizer": "CLIPTokenizer",
13
- "unet": "UNet2DConditionModel",
14
- "vae": "AutoencoderKL"
15
  }
 
1
  {
2
  "_class_name": "SuperDiffPipeline",
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
  }