insideman commited on
Commit
ea22ea4
·
verified ·
1 Parent(s): fac5946

Update model_index.json

Browse files
Files changed (1) hide show
  1. model_index.json +12 -5
model_index.json CHANGED
@@ -1,7 +1,16 @@
1
  {
2
- "_class_name": "StableDiffusionImg2ImgPipeline",
 
 
3
  "scheduler": {
4
- "_class_name": "DDIMScheduler"
 
 
 
 
 
 
 
5
  },
6
  "text_encoder": {
7
  "_class_name": "CLIPTextModel"
@@ -14,7 +23,5 @@
14
  },
15
  "vae": {
16
  "_class_name": "AutoencoderKL"
17
- },
18
- "requires_safety_checker": false,
19
- "safety_checker": null
20
  }
 
1
  {
2
+ "_class_name": "StableDiffusionPipeline",
3
+ "model_type": "stable-diffusion-v1-5",
4
+ "architectures": ["UNet2DConditionModel"],
5
  "scheduler": {
6
+ "_class_name": "PNDMScheduler",
7
+ "_diffusers_version": "0.21.4",
8
+ "beta_end": 0.012,
9
+ "beta_schedule": "scaled_linear",
10
+ "beta_start": 0.00085,
11
+ "num_train_timesteps": 1000,
12
+ "skip_prk_steps": true,
13
+ "steps_offset": 1
14
  },
15
  "text_encoder": {
16
  "_class_name": "CLIPTextModel"
 
23
  },
24
  "vae": {
25
  "_class_name": "AutoencoderKL"
26
+ }
 
 
27
  }