Method model_fn

ldm/models/diffusion/dpm_solver/dpm_solver.py:367–374

Convert the model to the noise prediction model or the data prediction model.

(self, x, t)

        return x0

    def model_fn(self, x, t):
        """
        Convert the model to the noise prediction model or the data prediction model.
        """
        if self.predict_x0:
            return self.data_prediction_fn(x, t)
        else:
            return self.noise_prediction_fn(x, t)

    def get_time_steps(self, skip_type, t_T, t_0, N, device):
        """Compute the intermediate time steps for sampling.