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

Update model_index.json

Browse files
Files changed (1) hide show
  1. model_index.json +11 -8
model_index.json CHANGED
@@ -1,12 +1,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
  }
 
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
  }