Add timestep_spacing and steps_offset to schedulers

src/diffusers/configuration_utils.py

@@ -508,7 +508,8 @@ def extract_init_dict(cls, config_dict, **kwargs):
         hidden_config_dict = {k: v for k, v in original_dict.items() if k not in init_dict}
 
         # 8. Take note of the parameters that were not present in the loaded config
-        hidden_config_dict["_use_default_values"] = expected_keys - set(init_dict)
+        if len(expected_keys - set(init_dict)) > 0:
+            hidden_config_dict["_use_default_values"] = expected_keys - set(init_dict)
 
         return init_dict, unused_kwargs, hidden_config_dict
 

src/diffusers/schedulers/scheduling_euler_discrete.py

@@ -146,9 +146,6 @@ def __init__(
         sigmas = np.concatenate([sigmas[::-1], [0.0]]).astype(np.float32)
         self.sigmas = torch.from_numpy(sigmas)
 
-        # standard deviation of the initial noise distribution
-        self.init_noise_sigma = self.sigmas.max()
-
         # setable values
         self.num_inference_steps = None
         timesteps = np.linspace(0, num_train_timesteps - 1, num_train_timesteps, dtype=float)[::-1].copy()

tests/others/test_config.py

@@ -75,6 +75,22 @@ def __init__(
         pass
 
 
+class SampleObject4(ConfigMixin):
+    config_name = "config.json"
+
+    @register_to_config
+    def __init__(
+        self,
+        a=2,
+        b=5,
+        c=(2, 5),
+        d="for diffusion",
+        e=[1, 5],
+        f=[5, 4],
+    ):
+        pass
+
+
 class ConfigTester(unittest.TestCase):
     def test_load_not_from_mixin(self):
         with self.assertRaises(ValueError):
@@ -233,3 +249,21 @@ def test_load_dpmsolver(self):
         assert dpm.__class__ == DPMSolverMultistepScheduler
         # no warning should be thrown
         assert cap_logger.out == ""
+
+    def test_use_default_values(self):
+        # let's first save a config that should be in the form
+        #    a=2,
+        #    b=5,
+        #    c=(2, 5),
+        #    d="for diffusion",
+        #    e=[1, 3],
+
+        config = SampleObject()
+        with tempfile.TemporaryDirectory() as tmpdirname:
+            config.save(tmpdirname)
+
+            # now loading it with SampleObject2 should put f into `_use_default_values`
+            config = SampleObject2.from_config(tmpdirname)
+
+            assert "f" in config._use_default_values
+            assert config.f == [1, 4]