dg845 commited on
Commit
323a9af
1 Parent(s): c7f9b67

Upload 2 files

Browse files

Add `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 CHANGED
@@ -5,13 +5,14 @@
5
  "transformers",
6
  "CLIPImageProcessor"
7
  ],
 
8
  "safety_checker": [
9
  "stable_diffusion",
10
  "StableDiffusionSafetyChecker"
11
  ],
12
  "scheduler": [
13
- null,
14
- null
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": true,
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,