Rafi Rizwan commited on
Commit
a7907bc
·
1 Parent(s): 4c386f8

disable v prediction

Browse files
Files changed (1) hide show
  1. scheduler/scheduler_config.json +1 -1
scheduler/scheduler_config.json CHANGED
@@ -8,7 +8,7 @@
8
  "clip_sample_range": 1.0,
9
  "dynamic_thresholding_ratio": 0.995,
10
  "num_train_timesteps": 1000,
11
- "prediction_type": "v_prediction",
12
  "sample_max_value": 1.0,
13
  "set_alpha_to_one": false,
14
  "skip_prk_steps": true,
 
8
  "clip_sample_range": 1.0,
9
  "dynamic_thresholding_ratio": 0.995,
10
  "num_train_timesteps": 1000,
11
+ "prediction_type": "epsilon",
12
  "sample_max_value": 1.0,
13
  "set_alpha_to_one": false,
14
  "skip_prk_steps": true,