Unverified Commit 53377ef8 authored by Nipun Jindal's avatar Nipun Jindal Committed by GitHub
Browse files

[2761]: Add documentation for extra_in_channels UNet1DModel (#2817)


Co-authored-by: default avatarnjindal <njindal@adobe.com>
parent 4d0f412d
......@@ -47,6 +47,9 @@ class UNet1DModel(ModelMixin, ConfigMixin):
sample_size (`int`, *optional*): Default length of sample. Should be adaptable at runtime.
in_channels (`int`, *optional*, defaults to 2): Number of channels in the input sample.
out_channels (`int`, *optional*, defaults to 2): Number of channels in the output.
extra_in_channels (`int`, *optional*, defaults to 0):
Number of additional channels to be added to the input of the first down block. Useful for cases where the
input data has more channels than what the model is initially designed for.
time_embedding_type (`str`, *optional*, defaults to `"fourier"`): Type of time embedding to use.
freq_shift (`float`, *optional*, defaults to 0.0): Frequency shift for fourier time embedding.
flip_sin_to_cos (`bool`, *optional*, defaults to :
......
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