update

canonicalize/models/unet_mv2d_blocks.py

@@ -21,7 +21,7 @@ from torch import nn
 from diffusers.utils import is_torch_version, logging
 # from diffusers.models.attention import AdaGroupNorm
 from diffusers.models.attention_processor import Attention, AttnAddedKVProcessor, AttnAddedKVProcessor2_0
-from diffusers.models.dual_transformer_2d import DualTransformer2DModel
+from diffusers.models.transformers.dual_transformer_2d import DualTransformer2DModel
 from diffusers.models.resnet import Downsample2D, FirDownsample2D, FirUpsample2D, KDownsample2D, KUpsample2D, ResnetBlock2D, Upsample2D
 from canonicalize.models.transformer_mv2d import TransformerMV2DModel
 

canonicalize/models/unet_mv2d_condition.py

@@ -38,7 +38,7 @@ from diffusers.models.embeddings import (
     Timesteps,
 )
 from diffusers.models.modeling_utils import ModelMixin, load_state_dict, _load_state_dict_into_model
-from diffusers.models.unet_2d_blocks import (
+from diffusers.models.unets.unet_2d_blocks import (
     CrossAttnDownBlock2D,
     CrossAttnUpBlock2D,
     DownBlock2D,

canonicalize/models/unet_mv2d_ref.py

@@ -40,7 +40,7 @@ from diffusers.models.embeddings import (
 from diffusers.models.lora import LoRALinearLayer
 
 from diffusers.models.modeling_utils import ModelMixin, load_state_dict, _load_state_dict_into_model
-from diffusers.models.unet_2d_blocks import (
+from diffusers.models.unets.unet_2d_blocks import (
     CrossAttnDownBlock2D,
     CrossAttnUpBlock2D,
     DownBlock2D,

multiview/models/unet_mv2d_blocks.py

@@ -21,7 +21,7 @@ from torch import nn
 from diffusers.utils import is_torch_version, logging
 from diffusers.models.normalization import AdaGroupNorm
 from diffusers.models.attention_processor import Attention, AttnAddedKVProcessor, AttnAddedKVProcessor2_0
-from diffusers.models.dual_transformer_2d import DualTransformer2DModel
+from diffusers.models.transformers.dual_transformer_2d import DualTransformer2DModel
 from diffusers.models.resnet import Downsample2D, FirDownsample2D, FirUpsample2D, KDownsample2D, KUpsample2D, ResnetBlock2D, Upsample2D
 
 from diffusers.models.unets.unet_2d_blocks import DownBlock2D, ResnetDownsampleBlock2D, AttnDownBlock2D, CrossAttnDownBlock2D, SimpleCrossAttnDownBlock2D, SkipDownBlock2D, AttnSkipDownBlock2D, DownEncoderBlock2D, AttnDownEncoderBlock2D, KDownBlock2D, KCrossAttnDownBlock2D

multiview/models/unet_mv2d_condition.py

@@ -36,7 +36,7 @@ from diffusers.models.embeddings import (
     Timesteps,
 )
 from diffusers.models.modeling_utils import ModelMixin, load_state_dict, _load_state_dict_into_model
-from diffusers.models.unet_2d_blocks import (
+from diffusers.models.unets.unet_2d_blocks import (
     CrossAttnDownBlock2D,
     CrossAttnUpBlock2D,
     DownBlock2D,

pre-requirements.txt

@@ -3,10 +3,10 @@ torchvision==0.16.0
 xformers==0.0.22.post7
 omegaconf
 rm_anime_bg
-diffusers==0.27.2
+diffusers==0.29.1
 transformers==4.42.4
 einops
-huggingface_hub==0.25.0
+huggingface_hub==0.29.3
 opencv-python
 accelerate
 matplotlib