mskrt commited on
Commit
64319ec
verified
1 Parent(s): 0b0d11b

Update model_index.json

Browse files
Files changed (1) hide show
  1. model_index.json +17 -12
model_index.json CHANGED
@@ -1,15 +1,20 @@
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
+ "scheduler": ["diffusers", "EulerDiscreteScheduler"],
7
+ "text_encoder": ["transformers", "CLIPTextModel"],
8
+ "tokenizer": ["transformers", "CLIPTokenizer"],
9
+ "unet": ["diffusers", "UNet2DConditionModel"],
10
+ "vae": ["diffusers", "AutoencoderKL"]
11
+ },
12
+ "default_config": {
13
+ "batch_size": 1,
14
+ "device": "cuda",
15
+ "guidance_scale": 7.5,
16
+ "lift": 0.0,
17
+ "num_inference_steps": 1000,
18
+ "seed": 1
19
+ }
20
+ }