mskrt commited on
Commit
3e9f243
verified
1 Parent(s): 094c541

Update model_index.json

Browse files
Files changed (1) hide show
  1. model_index.json +12 -9
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
+ }