From 520648b98a9932b4c5b0e411dd321e7791c89f3c Mon Sep 17 00:00:00 2001 From: Nathan Lambert Date: Mon, 10 Oct 2022 11:13:31 -0700 Subject: [PATCH] fix typo docstring --- src/diffusers/models/unet_2d_condition.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/unet_2d_condition.py b/src/diffusers/models/unet_2d_condition.py index d510589404bc..4e4eaddf5dfe 100644 --- a/src/diffusers/models/unet_2d_condition.py +++ b/src/diffusers/models/unet_2d_condition.py @@ -223,7 +223,7 @@ def forward( 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