rezard35 commited on
Commit
29a5290
1 Parent(s): 5cae40e

Update scheduler/scheduler_config.json

Browse files
Files changed (1) hide show
  1. scheduler/scheduler_config.json +1 -1
scheduler/scheduler_config.json CHANGED
@@ -5,7 +5,7 @@
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
  "clip_sample": false,
8
- "num_train_timesteps": 1000,
9
  "prediction_type": "v_prediction",
10
  "set_alpha_to_one": false,
11
  "skip_prk_steps": true,
 
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
  "clip_sample": false,
8
+ "num_train_timesteps": 500,
9
  "prediction_type": "v_prediction",
10
  "set_alpha_to_one": false,
11
  "skip_prk_steps": true,