Add `steps_offset=1` to config to suppress deprecation warning (#12)

Browse files

- Add `steps_offset=1` to config to suppress deprecation warning (b5f06b80285972d3455892ad13a0bd3f0dae8709)

Co-authored-by: Patrick von Platen <[email protected]>

Files changed (1) hide show

scheduler/scheduler_config.json +2 -1

scheduler/scheduler_config.json CHANGED Viewed

@@ -8,5 +8,6 @@
   "num_train_timesteps": 1000,
   "set_alpha_to_one": false,
   "timestep_values": null,
-  "trained_betas": null
 }

   "num_train_timesteps": 1000,
   "set_alpha_to_one": false,
   "timestep_values": null,
+  "trained_betas": null,
+  "steps_offset": 1
 }