-
Notifications
You must be signed in to change notification settings - Fork 6.5k
Closed
Labels
bugSomething isn't workingSomething isn't working
Description
Describe the bug
As seen here, the forward docstrings of the UNet classes have "r" as the method description. Why is this?
diffusers/src/diffusers/models/unet_2d.py
Lines 173 to 180 in 688031c
| """r | |
| Args: | |
| sample (`torch.FloatTensor`): (batch, channel, height, width) 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_2d.UNet2DOutput`] instead of a plain tuple. | |
| Returns: |
I think @kashif you added these?
Reproduction
No response
Logs
No response
System Info
Not needed, documentation issue.
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't working