Skip to content

Commit 938ec58

Browse files
nipunjindalnjindal
and
njindal
authored
[2761]: Add documentation for extra_in_channels UNet1DModel (huggingface#2817)
Co-authored-by: njindal <[email protected]>
1 parent 5181972 commit 938ec58

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

models/unet_1d.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,9 @@ class UNet1DModel(ModelMixin, ConfigMixin):
4747
sample_size (`int`, *optional*): Default length of sample. Should be adaptable at runtime.
4848
in_channels (`int`, *optional*, defaults to 2): Number of channels in the input sample.
4949
out_channels (`int`, *optional*, defaults to 2): Number of channels in the output.
50+
extra_in_channels (`int`, *optional*, defaults to 0):
51+
Number of additional channels to be added to the input of the first down block. Useful for cases where the
52+
input data has more channels than what the model is initially designed for.
5053
time_embedding_type (`str`, *optional*, defaults to `"fourier"`): Type of time embedding to use.
5154
freq_shift (`float`, *optional*, defaults to 0.0): Frequency shift for fourier time embedding.
5255
flip_sin_to_cos (`bool`, *optional*, defaults to :

0 commit comments

Comments
 (0)