From f1b9ee7ed9af31f942308e3e6ac5d35411d40183 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Tue, 4 Oct 2022 15:06:11 +0200 Subject: [PATCH] [Docs] fix docstring for issue #709 (#710) fix docstring fixes #709 --- src/diffusers/models/unet_2d.py | 2 +- src/diffusers/models/unet_2d_condition_flax.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/models/unet_2d.py b/src/diffusers/models/unet_2d.py index 89321a55..2415bf4e 100644 --- a/src/diffusers/models/unet_2d.py +++ b/src/diffusers/models/unet_2d.py @@ -170,7 +170,7 @@ class UNet2DModel(ModelMixin, ConfigMixin): timestep: Union[torch.Tensor, float, int], return_dict: bool = True, ) -> Union[UNet2DOutput, Tuple]: - """r + r""" Args: sample (`torch.FloatTensor`): (batch, channel, height, width) noisy inputs tensor timestep (`torch.FloatTensor` or `float` or `int): (batch) timesteps diff --git a/src/diffusers/models/unet_2d_condition_flax.py b/src/diffusers/models/unet_2d_condition_flax.py index 9e84da06..6c83bbdb 100644 --- a/src/diffusers/models/unet_2d_condition_flax.py +++ b/src/diffusers/models/unet_2d_condition_flax.py @@ -215,7 +215,7 @@ class FlaxUNet2DConditionModel(nn.Module, FlaxModelMixin, ConfigMixin): return_dict: bool = True, train: bool = False, ) -> Union[FlaxUNet2DConditionOutput, Tuple]: - """r + r""" Args: sample (`jnp.ndarray`): (channel, height, width) noisy inputs tensor timestep (`jnp.ndarray` or `float` or `int`): timesteps