Skip to content

Commit 4488713

Browse files
SammyAgrawalDN6
andauthored
Update src/diffusers/models/unets/unet_1d.py
Co-authored-by: Dhruv Nair <[email protected]>
1 parent 7881392 commit 4488713

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/unets/unet_1d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def __init__(
8282
out_channels: int = 2,
8383
extra_in_channels: int = 0,
8484
time_embedding_type: str = "fourier",
85-
time_embedding_dim: int = 0,
85+
time_embedding_dim: Optional[int] = None,
8686
flip_sin_to_cos: bool = True,
8787
use_timestep_embedding: bool = False,
8888
freq_shift: float = 0.0,

0 commit comments

Comments
 (0)