up

model_index.json

@@ -11,7 +11,7 @@
   ],
   "scheduler": [
     "diffusers",
-    "PNDMScheduler"
+    "FlaxDDIMScheduler"
   ],
   "text_encoder": [
     "transformers",

scheduler/scheduler_config.json

@@ -1,12 +1,11 @@
 {
-  "_class_name": "PNDMScheduler",
-  "_diffusers_version": "0.1.3",
+  "_class_name": "FlaxDDIMScheduler",
+  "_diffusers_version": "0.4.0.dev0",
   "beta_end": 0.012,
   "beta_schedule": "scaled_linear",
   "beta_start": 0.00085,
-  "clip_sample": true,
+  "clip_sample": false,
   "num_train_timesteps": 1000,
-  "timestep_values": null,
-  "skip_prk_steps": true,
-  "trained_betas": null
-}
+  "set_alpha_to_one": false,
+  "steps_offset": 1
+}