1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

fix typo docstring in unet2d (#798)

fix typo docstring
This commit is contained in:
Nathan Lambert
2022-10-10 11:25:20 -07:00
committed by GitHub
parent 22963ed826
commit 71ca10c6a4

View File

@@ -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