File tree 1 file changed +3
-0
lines changed 1 file changed +3
-0
lines changed Original file line number Diff line number Diff line change @@ -47,6 +47,9 @@ class UNet1DModel(ModelMixin, ConfigMixin):
47
47
sample_size (`int`, *optional*): Default length of sample. Should be adaptable at runtime.
48
48
in_channels (`int`, *optional*, defaults to 2): Number of channels in the input sample.
49
49
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.
50
53
time_embedding_type (`str`, *optional*, defaults to `"fourier"`): Type of time embedding to use.
51
54
freq_shift (`float`, *optional*, defaults to 0.0): Frequency shift for fourier time embedding.
52
55
flip_sin_to_cos (`bool`, *optional*, defaults to :
You can’t perform that action at this time.
0 commit comments