patrickvonplaten commited on
Commit
c3bd02c
1 Parent(s): 4de6ae8

Add `clip_sample=False` to scheduler to make model compatible with DDIM.

Browse files

Hey sd-dreambooth-library 👋,

Your model repository seems to contain a stable diffusion checkpoint. We have noticed that your scheduler config currently does not correctly work with the [DDIMScheduler](https://huggingface.co/docs/diffusers/main/en/api/schedulers#diffusers.DDIMScheduler) because `clip_sample` is not set to False and will therefore [incorrectly default to True](https://github.com/huggingface/diffusers/blob/3ce6380d3a2ec5c3e3f4f48889d380d657b151bc/src/diffusers/schedulers/scheduling_ddim.py#L127).
The official stable diffusion checkpoints have `clip_sample=False` so that the scheduler config works will **all** schedulers, see: https://huggingface.co/stabilityai/stable-diffusion-2-1-base/blob/main/scheduler/scheduler_config.json#L7.

We strongly recommend that you merge this PR to make sure your model works correctly with DDIM.

Diffusingly,
Patrick.

Files changed (1) hide show
  1. scheduler/scheduler_config.json +1 -0
scheduler/scheduler_config.json CHANGED
@@ -4,6 +4,7 @@
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
 
7
  "num_train_timesteps": 1000,
8
  "set_alpha_to_one": false,
9
  "skip_prk_steps": true,
 
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
+ "clip_sample": false,
8
  "num_train_timesteps": 1000,
9
  "set_alpha_to_one": false,
10
  "skip_prk_steps": true,