clip_sample

Author: hlky

src/diffusers/schedulers/schedules/beta_schedule.py

@@ -104,8 +104,6 @@ def __init__(
         timestep_spacing: str = "linspace",
         timestep_type: str = "discrete",  # can be "discrete" or "continuous"
         steps_offset: int = 0,
-        clip_sample: Optional[bool] = False,
-        clip_sample_range: float = 1.0,
         sigma_min: Optional[float] = None,
         sigma_max: Optional[float] = None,
         final_sigmas_type: str = "zero",  # can be "zero" or "sigma_min"
@@ -145,8 +143,6 @@ def __init__(
         self.timestep_spacing = timestep_spacing
         self.timestep_type = timestep_type
         self.steps_offset = steps_offset
-        self.clip_sample = clip_sample
-        self.clip_sample_range = clip_sample_range
         self.sigma_min = sigma_min
         self.sigma_max = sigma_max
         self.final_sigmas_type = final_sigmas_type

src/diffusers/schedulers/scheduling_heun_discrete.py

@@ -85,6 +85,8 @@ def __init__(
         schedule_config,
         sigma_schedule_config,
         prediction_type: str = "epsilon",
+        clip_sample: Optional[bool] = False,
+        clip_sample_range: float = 1.0,
     ):
         self.set_schedule(schedule_config)
         self.set_sigma_schedule(sigma_schedule_config)
@@ -264,9 +266,9 @@ def step(
                 f"prediction_type given as {self.config.prediction_type} must be one of `epsilon`, or `v_prediction`"
             )
 
-        if self._schedule.clip_sample:
+        if self.config.clip_sample:
             pred_original_sample = pred_original_sample.clamp(
-                -self._schedule.clip_sample_range, self._schedule.clip_sample_range
+                -self.config.clip_sample_range, self.config.clip_sample_range
             )
 
         if self.state_in_first_order: