Method p_mean_variance

diff2flow/openai_diffusion/respace.py:89–92

(
        self, model, *args, **kwargs
    )

        super().__init__(**kwargs)

    def p_mean_variance(
        self, model, *args, **kwargs
    ):  # pylint: disable=signature-differs
        return super().p_mean_variance(self._wrap_model(model), *args, **kwargs)

    def training_losses(
        self, model, *args, **kwargs