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

[Unet1d] correct docs (#2565)

Browse Source
This commit is contained in:
Patrick von Platen
2023-03-06 14:36:28 +01:00
committed by GitHub
parent 1598a57958
commit ec021923d2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/models/unet_1d.py
View File

@@ -195,7 +195,7 @@ class UNet1DModel(ModelMixin, ConfigMixin):
) -> Union[UNet1DOutput, Tuple]:
r"""
Args:
sample (`torch.FloatTensor`): `(batch_size, sample_size, num_channels)` noisy inputs tensor
sample (`torch.FloatTensor`): `(batch_size, num_channels, sample_size)` noisy inputs tensor
timestep (`torch.FloatTensor` or `float` or `int): (batch) timesteps
return_dict (`bool`, *optional*, defaults to `True`):
Whether or not to return a [`~models.unet_1d.UNet1DOutput`] instead of a plain tuple.