From 68fa3fed5753cc67f880968633f9a19b0a634bfa Mon Sep 17 00:00:00 2001
From: Your Name <your.email@example.com>
Date: Wed, 12 Nov 2025 09:07:37 +0000
Subject: [PATCH] Update UNet models for CharacterGen, Era3D, and Wonder3D

---
 .../Stage_2D/tuneavideo/models/unet_mv2d_condition.py          | 3 ++-
 .../CharacterGen/Stage_2D/tuneavideo/models/unet_mv2d_ref.py   | 3 ++-
 Gen_3D_Modules/Era3D/mvdiffusion/models/unet_mv2d_condition.py | 3 ++-
 Gen_3D_Modules/Wonder3D/models/unet_mv2d_condition.py          | 3 ++-
 4 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/Gen_3D_Modules/CharacterGen/Stage_2D/tuneavideo/models/unet_mv2d_condition.py b/Gen_3D_Modules/CharacterGen/Stage_2D/tuneavideo/models/unet_mv2d_condition.py
index 3f787c3..cca62d9 100644
--- a/Gen_3D_Modules/CharacterGen/Stage_2D/tuneavideo/models/unet_mv2d_condition.py
+++ b/Gen_3D_Modules/CharacterGen/Stage_2D/tuneavideo/models/unet_mv2d_condition.py
@@ -37,7 +37,8 @@ from diffusers.models.embeddings import (
     TimestepEmbedding,
     Timesteps,
 )
-from diffusers.models.modeling_utils import ModelMixin, load_state_dict, _load_state_dict_into_model
+from diffusers.models.modeling_utils import ModelMixin
+from diffusers.models.model_loading_utils import load_state_dict, _load_state_dict_into_model
 from diffusers.models.unets.unet_2d_blocks import (
     CrossAttnDownBlock2D,
     CrossAttnUpBlock2D,
diff --git a/Gen_3D_Modules/CharacterGen/Stage_2D/tuneavideo/models/unet_mv2d_ref.py b/Gen_3D_Modules/CharacterGen/Stage_2D/tuneavideo/models/unet_mv2d_ref.py
index c5c9316..466e7d1 100644
--- a/Gen_3D_Modules/CharacterGen/Stage_2D/tuneavideo/models/unet_mv2d_ref.py
+++ b/Gen_3D_Modules/CharacterGen/Stage_2D/tuneavideo/models/unet_mv2d_ref.py
@@ -39,7 +39,8 @@ 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.modeling_utils import ModelMixin
+from diffusers.models.model_loading_utils import load_state_dict, _load_state_dict_into_model
 from diffusers.models.unets.unet_2d_blocks import (
     CrossAttnDownBlock2D,
     CrossAttnUpBlock2D,
diff --git a/Gen_3D_Modules/Era3D/mvdiffusion/models/unet_mv2d_condition.py b/Gen_3D_Modules/Era3D/mvdiffusion/models/unet_mv2d_condition.py
index e62845e..a0213e9 100644
--- a/Gen_3D_Modules/Era3D/mvdiffusion/models/unet_mv2d_condition.py
+++ b/Gen_3D_Modules/Era3D/mvdiffusion/models/unet_mv2d_condition.py
@@ -35,7 +35,8 @@ from diffusers.models.embeddings import (
     TimestepEmbedding,
     Timesteps,
 )
-from diffusers.models.modeling_utils import ModelMixin, load_state_dict, _load_state_dict_into_model
+from diffusers.models.modeling_utils import ModelMixin
+from diffusers.models.model_loading_utils import load_state_dict, _load_state_dict_into_model
 from diffusers.models.unets.unet_2d_blocks import (
     CrossAttnDownBlock2D,
     CrossAttnUpBlock2D,
diff --git a/Gen_3D_Modules/Wonder3D/models/unet_mv2d_condition.py b/Gen_3D_Modules/Wonder3D/models/unet_mv2d_condition.py
index f530c4a..b87422b 100644
--- a/Gen_3D_Modules/Wonder3D/models/unet_mv2d_condition.py
+++ b/Gen_3D_Modules/Wonder3D/models/unet_mv2d_condition.py
@@ -35,7 +35,8 @@ from diffusers.models.embeddings import (
     TimestepEmbedding,
     Timesteps,
 )
-from diffusers.models.modeling_utils import ModelMixin, load_state_dict, _load_state_dict_into_model
+from diffusers.models.modeling_utils import ModelMixin
+from diffusers.models.model_loading_utils import load_state_dict, _load_state_dict_into_model
 from diffusers.models.unets.unet_2d_blocks import (
     CrossAttnDownBlock2D,
     CrossAttnUpBlock2D,