cats / scheduler /scheduler_config.json
nroggendorff's picture
Rename scheduler/config.json to scheduler/scheduler_config.json
cae3c0d verified
{
"num_train_timesteps": 1000,
"beta_start": 0.0001,
"beta_end": 0.02,
"beta_schedule": "linear"
}