Upload 2 files
Browse filesAdd `model_index.json` and updated scheduler config to allow loading from `diffusers` pipeline. (Should probably only merge after PR #5448 is merged to `diffusers` since the scheduler config change might not be compatible with the community pipeline.)
- model_index.json +3 -2
- scheduler/scheduler_config.json +2 -1
model_index.json
CHANGED
@@ -5,13 +5,14 @@
|
|
5 |
"transformers",
|
6 |
"CLIPImageProcessor"
|
7 |
],
|
|
|
8 |
"safety_checker": [
|
9 |
"stable_diffusion",
|
10 |
"StableDiffusionSafetyChecker"
|
11 |
],
|
12 |
"scheduler": [
|
13 |
-
|
14 |
-
|
15 |
],
|
16 |
"text_encoder": [
|
17 |
"transformers",
|
|
|
5 |
"transformers",
|
6 |
"CLIPImageProcessor"
|
7 |
],
|
8 |
+
"requires_safety_checker": true,
|
9 |
"safety_checker": [
|
10 |
"stable_diffusion",
|
11 |
"StableDiffusionSafetyChecker"
|
12 |
],
|
13 |
"scheduler": [
|
14 |
+
"diffusers",
|
15 |
+
"LCMScheduler"
|
16 |
],
|
17 |
"text_encoder": [
|
18 |
"transformers",
|
scheduler/scheduler_config.json
CHANGED
@@ -4,10 +4,11 @@
|
|
4 |
"beta_end": 0.012,
|
5 |
"beta_schedule": "scaled_linear",
|
6 |
"beta_start": 0.00085,
|
7 |
-
"clip_sample":
|
8 |
"clip_sample_range": 1.0,
|
9 |
"dynamic_thresholding_ratio": 0.995,
|
10 |
"num_train_timesteps": 1000,
|
|
|
11 |
"prediction_type": "epsilon",
|
12 |
"rescale_betas_zero_snr": false,
|
13 |
"sample_max_value": 1.0,
|
|
|
4 |
"beta_end": 0.012,
|
5 |
"beta_schedule": "scaled_linear",
|
6 |
"beta_start": 0.00085,
|
7 |
+
"clip_sample": false,
|
8 |
"clip_sample_range": 1.0,
|
9 |
"dynamic_thresholding_ratio": 0.995,
|
10 |
"num_train_timesteps": 1000,
|
11 |
+
"original_inference_steps": 50,
|
12 |
"prediction_type": "epsilon",
|
13 |
"rescale_betas_zero_snr": false,
|
14 |
"sample_max_value": 1.0,
|