Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion animation/modules/unet.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from diffusers.models.modeling_utils import ModelMixin
from diffusers.utils import BaseOutput, logging

from animation.modules.unet_3d_blocks import get_down_block, UNetMidBlockSpatioTemporal, get_up_block
from .unet_3d_blocks import get_down_block, UNetMidBlockSpatioTemporal, get_up_block
# from diffusers.models.unets.unet_3d_blocks import get_down_block, get_up_block, UNetMidBlockSpatioTemporal


Expand Down
2 changes: 1 addition & 1 deletion animation/modules/unet_3d_blocks.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@
# TransformerSpatioTemporalModel,
# TransformerTemporalModel,
# )
from animation.modules.transformer_temporal import TransformerTemporalModel, TransformerSpatioTemporalModel
from .transformer_temporal import TransformerTemporalModel, TransformerSpatioTemporalModel

from diffusers.models.unets.unet_motion_model import (
CrossAttnDownBlockMotion,
Expand Down
2 changes: 1 addition & 1 deletion animation/pipelines/inference_pipeline_animation.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
from diffusers.utils import BaseOutput, logging
from diffusers.utils.torch_utils import is_compiled_module, randn_tensor

from animation.modules.attention_processor import AnimationAttnProcessor, AnimationIDAttnProcessor
from ..modules.attention_processor import AnimationAttnProcessor, AnimationIDAttnProcessor
from einops import rearrange

logger = logging.get_logger(__name__) # pylint: disable=invalid-name
Expand Down