Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
chenpangpang
diffusers
Commits
3a28e36a
Unverified
Commit
3a28e36a
authored
May 29, 2024
by
Sayak Paul
Committed by
GitHub
May 29, 2024
Browse files
[Post release 0.28.0] remove deprecated blocks. (#8291)
* remove deprecated blocks. * update the location paths.
parent
3393c01c
Changes
24
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
15 additions
and
15 deletions
+15
-15
src/diffusers/models/unets/unet_3d_condition.py
src/diffusers/models/unets/unet_3d_condition.py
+4
-4
src/diffusers/models/unets/unet_i2vgen_xl.py
src/diffusers/models/unets/unet_i2vgen_xl.py
+4
-4
src/diffusers/models/unets/unet_motion_model.py
src/diffusers/models/unets/unet_motion_model.py
+4
-4
src/diffusers/models/unets/unet_spatio_temporal_condition.py
src/diffusers/models/unets/unet_spatio_temporal_condition.py
+3
-3
No files found.
src/diffusers/models/unets/unet_3d_condition.py
View file @
3a28e36a
...
@@ -598,15 +598,15 @@ class UNet3DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixin)
...
@@ -598,15 +598,15 @@ class UNet3DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixin)
mid_block_additional_residual: (`torch.Tensor`, *optional*):
mid_block_additional_residual: (`torch.Tensor`, *optional*):
A tensor that if specified is added to the residual of the middle unet block.
A tensor that if specified is added to the residual of the middle unet block.
return_dict (`bool`, *optional*, defaults to `True`):
return_dict (`bool`, *optional*, defaults to `True`):
Whether or not to return a [`~models.unet_3d_condition.UNet3DConditionOutput`] instead of a plain
Whether or not to return a [`~models.
unets.
unet_3d_condition.UNet3DConditionOutput`] instead of a plain
tuple.
tuple.
cross_attention_kwargs (`dict`, *optional*):
cross_attention_kwargs (`dict`, *optional*):
A kwargs dictionary that if specified is passed along to the [`AttnProcessor`].
A kwargs dictionary that if specified is passed along to the [`AttnProcessor`].
Returns:
Returns:
[`~models.unet_3d_condition.UNet3DConditionOutput`] or `tuple`:
[`~models.
unets.
unet_3d_condition.UNet3DConditionOutput`] or `tuple`:
If `return_dict` is True, an [`~models.unet_3d_condition.UNet3DConditionOutput`] is returned,
otherwise
If `return_dict` is True, an [`~models.
unets.
unet_3d_condition.UNet3DConditionOutput`] is returned,
a `tuple` is returned where the first element is the sample tensor.
otherwise
a `tuple` is returned where the first element is the sample tensor.
"""
"""
# By default samples have to be AT least a multiple of the overall upsampling factor.
# By default samples have to be AT least a multiple of the overall upsampling factor.
# The overall upsampling factor is equal to 2 ** (# num of upsampling layears).
# The overall upsampling factor is equal to 2 ** (# num of upsampling layears).
...
...
src/diffusers/models/unets/unet_i2vgen_xl.py
View file @
3a28e36a
...
@@ -542,13 +542,13 @@ class I2VGenXLUNet(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixin):
...
@@ -542,13 +542,13 @@ class I2VGenXLUNet(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixin):
`self.processor` in
`self.processor` in
[diffusers.models.attention_processor](https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/attention_processor.py).
[diffusers.models.attention_processor](https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/attention_processor.py).
return_dict (`bool`, *optional*, defaults to `True`):
return_dict (`bool`, *optional*, defaults to `True`):
Whether or not to return a [`~models.unet_3d_condition.UNet3DConditionOutput`] instead of a plain
Whether or not to return a [`~models.
unets.
unet_3d_condition.UNet3DConditionOutput`] instead of a plain
tuple.
tuple.
Returns:
Returns:
[`~models.unet_3d_condition.UNet3DConditionOutput`] or `tuple`:
[`~models.
unets.
unet_3d_condition.UNet3DConditionOutput`] or `tuple`:
If `return_dict` is True, an [`~models.unet_3d_condition.UNet3DConditionOutput`] is returned,
otherwise
If `return_dict` is True, an [`~models.
unets.
unet_3d_condition.UNet3DConditionOutput`] is returned,
a `tuple` is returned where the first element is the sample tensor.
otherwise
a `tuple` is returned where the first element is the sample tensor.
"""
"""
batch_size
,
channels
,
num_frames
,
height
,
width
=
sample
.
shape
batch_size
,
channels
,
num_frames
,
height
,
width
=
sample
.
shape
...
...
src/diffusers/models/unets/unet_motion_model.py
View file @
3a28e36a
...
@@ -856,13 +856,13 @@ class UNetMotionModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixin):
...
@@ -856,13 +856,13 @@ class UNetMotionModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixin):
mid_block_additional_residual: (`torch.Tensor`, *optional*):
mid_block_additional_residual: (`torch.Tensor`, *optional*):
A tensor that if specified is added to the residual of the middle unet block.
A tensor that if specified is added to the residual of the middle unet block.
return_dict (`bool`, *optional*, defaults to `True`):
return_dict (`bool`, *optional*, defaults to `True`):
Whether or not to return a [`~models.unet_3d_condition.UNet3DConditionOutput`] instead of a plain
Whether or not to return a [`~models.
unets.
unet_3d_condition.UNet3DConditionOutput`] instead of a plain
tuple.
tuple.
Returns:
Returns:
[`~models.unet_3d_condition.UNet3DConditionOutput`] or `tuple`:
[`~models.
unets.
unet_3d_condition.UNet3DConditionOutput`] or `tuple`:
If `return_dict` is True, an [`~models.unet_3d_condition.UNet3DConditionOutput`] is returned,
otherwise
If `return_dict` is True, an [`~models.
unets.
unet_3d_condition.UNet3DConditionOutput`] is returned,
a `tuple` is returned where the first element is the sample tensor.
otherwise
a `tuple` is returned where the first element is the sample tensor.
"""
"""
# By default samples have to be AT least a multiple of the overall upsampling factor.
# By default samples have to be AT least a multiple of the overall upsampling factor.
# The overall upsampling factor is equal to 2 ** (# num of upsampling layears).
# The overall upsampling factor is equal to 2 ** (# num of upsampling layears).
...
...
src/diffusers/models/unets/unet_spatio_temporal_condition.py
View file @
3a28e36a
...
@@ -57,9 +57,9 @@ class UNetSpatioTemporalConditionModel(ModelMixin, ConfigMixin, UNet2DConditionL
...
@@ -57,9 +57,9 @@ class UNetSpatioTemporalConditionModel(ModelMixin, ConfigMixin, UNet2DConditionL
The dimension of the cross attention features.
The dimension of the cross attention features.
transformer_layers_per_block (`int`, `Tuple[int]`, or `Tuple[Tuple]` , *optional*, defaults to 1):
transformer_layers_per_block (`int`, `Tuple[int]`, or `Tuple[Tuple]` , *optional*, defaults to 1):
The number of transformer blocks of type [`~models.attention.BasicTransformerBlock`]. Only relevant for
The number of transformer blocks of type [`~models.attention.BasicTransformerBlock`]. Only relevant for
[`~models.unet_3d_blocks.CrossAttnDownBlockSpatioTemporal`],
[`~models.
unets.
unet_3d_blocks.CrossAttnDownBlockSpatioTemporal`],
[`~models.unet_3d_blocks.CrossAttnUpBlockSpatioTemporal`],
[`~models.
unets.
unet_3d_blocks.CrossAttnUpBlockSpatioTemporal`],
[`~models.unet_3d_blocks.UNetMidBlockSpatioTemporal`].
[`~models.
unets.
unet_3d_blocks.UNetMidBlockSpatioTemporal`].
num_attention_heads (`int`, `Tuple[int]`, defaults to `(5, 10, 10, 20)`):
num_attention_heads (`int`, `Tuple[int]`, defaults to `(5, 10, 10, 20)`):
The number of attention heads.
The number of attention heads.
dropout (`float`, *optional*, defaults to 0.0): The dropout probability to use.
dropout (`float`, *optional*, defaults to 0.0): The dropout probability to use.
...
...
Prev
1
2
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment