insideman commited on
Commit
eba80d9
·
verified ·
1 Parent(s): 7221ef0

Update model_index.json

Browse files
Files changed (1) hide show
  1. model_index.json +26 -7
model_index.json CHANGED
@@ -1,10 +1,29 @@
1
  {
2
- "pipeline": "StableDiffusionPipeline",
3
  "_class_name": "StableDiffusionPipeline",
4
- "scheduler": ["DDIMScheduler", "PNDMScheduler"],
5
- "torch_dtype": "float16",
6
- "text_encoder": ["CLIPTextModel"],
7
- "tokenizer": ["CLIPTokenizer"],
8
- "unet": ["UNet2DConditionModel"],
9
- "vae": ["AutoencoderKL"]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  }
 
1
  {
 
2
  "_class_name": "StableDiffusionPipeline",
3
+ "scheduler": {
4
+ "_class_name": "DDIMScheduler",
5
+ "_diffusers_version": "0.21.4",
6
+ "beta_end": 0.012,
7
+ "beta_schedule": "scaled_linear",
8
+ "beta_start": 0.00085,
9
+ "clip_sample": false,
10
+ "num_train_timesteps": 1000,
11
+ "prediction_type": "epsilon",
12
+ "set_alpha_to_one": false,
13
+ "skip_prk_steps": true,
14
+ "steps_offset": 1,
15
+ "trained_betas": null
16
+ },
17
+ "text_encoder": {
18
+ "_class_name": "CLIPTextModel"
19
+ },
20
+ "tokenizer": {
21
+ "_class_name": "CLIPTokenizer"
22
+ },
23
+ "unet": {
24
+ "_class_name": "UNet2DConditionModel"
25
+ },
26
+ "vae": {
27
+ "_class_name": "AutoencoderKL"
28
+ }
29
  }