fix typo docstring in unet2d (#798)

fix typo docstring
2022-10-10 11:25:20 -07:00
parent 22963ed826
commit 71ca10c6a4
1 changed files with 1 additions and 1 deletions
@@ -223,7 +223,7 @@ class UNet2DConditionModel(ModelMixin, ConfigMixin):
        encoder_hidden_states: torch.Tensor,
        return_dict: bool = True,
    ) -> Union[UNet2DConditionOutput, Tuple]:
-        """r
+        r"""
        Args:
            sample (`torch.FloatTensor`): (batch, channel, height, width) noisy inputs tensor
            timestep (`torch.FloatTensor` or `float` or `int`): (batch) timesteps