Update model_index.json
Browse files- model_index.json +12 -5
model_index.json
CHANGED
@@ -1,7 +1,16 @@
|
|
1 |
{
|
2 |
-
"_class_name": "
|
|
|
|
|
3 |
"scheduler": {
|
4 |
-
"_class_name": "
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
}
|