Karumoon commited on
Commit
ec36a99
1 Parent(s): 90f2035

Update scheduler/scheduler_config.json

Browse files
Files changed (1) hide show
  1. scheduler/scheduler_config.json +2 -2
scheduler/scheduler_config.json CHANGED
@@ -4,10 +4,10 @@
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
- "num_train_timesteps": 1000,
8
  "prediction_type": "epsilon",
9
  "set_alpha_to_one": false,
10
  "skip_prk_steps": true,
11
- "steps_offset": 1,
12
  "trained_betas": null
13
  }
 
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
+ "num_train_timesteps": 10000,
8
  "prediction_type": "epsilon",
9
  "set_alpha_to_one": false,
10
  "skip_prk_steps": true,
11
+ "steps_offset": 2,
12
  "trained_betas": null
13
  }