Create model_index.json
Browse files- model_index.json +44 -0
model_index.json
ADDED
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "StableDiffusionPipeline",
|
3 |
+
"custom_pipeline": "stable_diffusion_text2img",
|
4 |
+
"feature_extractor": {
|
5 |
+
"_class_name": "CLIPFeatureExtractor",
|
6 |
+
"_diffusers_version": "0.21.4",
|
7 |
+
"do_normalize": true,
|
8 |
+
"do_resize": true,
|
9 |
+
"feature_extractor_type": "CLIPFeatureExtractor",
|
10 |
+
"image_mean": [0.48145466, 0.4578275, 0.40821073],
|
11 |
+
"image_std": [0.26862954, 0.26130258, 0.27577711],
|
12 |
+
"resample": 3,
|
13 |
+
"size": 224
|
14 |
+
},
|
15 |
+
"requires_safety_checker": false,
|
16 |
+
"safety_checker": null,
|
17 |
+
"scheduler": {
|
18 |
+
"_class_name": "DDIMScheduler",
|
19 |
+
"_diffusers_version": "0.21.4",
|
20 |
+
"beta_end": 0.012,
|
21 |
+
"beta_schedule": "scaled_linear",
|
22 |
+
"beta_start": 0.00085,
|
23 |
+
"clip_sample": false,
|
24 |
+
"num_train_timesteps": 1000,
|
25 |
+
"prediction_type": "epsilon",
|
26 |
+
"set_alpha_to_one": false,
|
27 |
+
"skip_prk_steps": true,
|
28 |
+
"steps_offset": 1,
|
29 |
+
"trained_betas": null
|
30 |
+
},
|
31 |
+
"text_encoder": {
|
32 |
+
"_class_name": "CLIPTextModel"
|
33 |
+
},
|
34 |
+
"tokenizer": {
|
35 |
+
"_class_name": "CLIPTokenizer"
|
36 |
+
},
|
37 |
+
"torch_dtype": "float16",
|
38 |
+
"unet": {
|
39 |
+
"_class_name": "UNet2DConditionModel"
|
40 |
+
},
|
41 |
+
"vae": {
|
42 |
+
"_class_name": "AutoencoderKL"
|
43 |
+
}
|
44 |
+
}
|