"git@developer.sourcefind.cn:gaoqiong/yaml-cpp.git" did not exist on "bd7f8c60c82614bb0bd1c526db2cbc39dac02fec"
Unverified Commit 71ca10c6 authored by Nathan Lambert's avatar Nathan Lambert Committed by GitHub
Browse files

fix typo docstring in unet2d (#798)

fix typo docstring
parent 22963ed8
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment