Upload folder using huggingface_hub

2D_Stage/tuneavideo/models/transformer_mv2d.py

@@ -27,7 +27,12 @@ except:
     from diffusers.utils.torch_utils import maybe_allow_in_graph
 from diffusers.models.attention import FeedForward, AdaLayerNorm, AdaLayerNormZero, Attention
 from diffusers.models.embeddings import PatchEmbed
-from diffusers.models.lora import LoRACompatibleConv, LoRACompatibleLinear
+try:
+    from diffusers.models.lora import LoRACompatibleConv, LoRACompatibleLinear
+except ImportError:
+    # Removed in newer diffusers; plain nn layers are equivalent during inference
+    LoRACompatibleConv = nn.Conv2d
+    LoRACompatibleLinear = nn.Linear
 from diffusers.models.modeling_utils import ModelMixin
 from diffusers.utils.import_utils import is_xformers_available
 

2D_Stage/tuneavideo/models/unet_mv2d_condition.py

@@ -37,7 +37,11 @@ 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
+try:
+    from diffusers.models.modeling_utils import load_state_dict, _load_state_dict_into_model
+except ImportError:
+    from diffusers.models.model_loading_utils import load_state_dict, _load_state_dict_into_model
 from diffusers.models.unet_2d_blocks import (
     CrossAttnDownBlock2D,
     CrossAttnUpBlock2D,
@@ -48,18 +52,40 @@ from diffusers.models.unet_2d_blocks import (
 )
 from diffusers.utils import (
     CONFIG_NAME,
-    DIFFUSERS_CACHE,
     FLAX_WEIGHTS_NAME,
-    HF_HUB_OFFLINE,
     SAFETENSORS_WEIGHTS_NAME,
     WEIGHTS_NAME,
-    _add_variant,
-    _get_model_file,
     deprecate,
     is_accelerate_available,
     is_torch_version,
     logging,
 )
+# DIFFUSERS_CACHE: removed in newer diffusers, replaced by huggingface_hub constant
+try:
+    from diffusers.utils import DIFFUSERS_CACHE
+except ImportError:
+    try:
+        from huggingface_hub.constants import HF_HUB_CACHE as DIFFUSERS_CACHE
+    except ImportError:
+        import os as _os
+        DIFFUSERS_CACHE = _os.path.join(_os.path.expanduser("~"), ".cache", "huggingface", "hub")
+# HF_HUB_OFFLINE: moved to huggingface_hub.constants in newer diffusers
+try:
+    from diffusers.utils import HF_HUB_OFFLINE
+except ImportError:
+    try:
+        from huggingface_hub.constants import HF_HUB_OFFLINE
+    except ImportError:
+        import os as _os
+        HF_HUB_OFFLINE = _os.environ.get("HF_HUB_OFFLINE", "0") == "1"
+# _add_variant, _get_model_file: may be in hub_utils in newer diffusers
+try:
+    from diffusers.utils import _add_variant, _get_model_file
+except ImportError:
+    try:
+        from diffusers.utils.hub_utils import _add_variant, _get_model_file
+    except ImportError:
+        from diffusers.utils.dynamic_modules_utils import _add_variant, _get_model_file
 from diffusers import __version__
 from tuneavideo.models.unet_mv2d_blocks import (
     CrossAttnDownBlockMV2D,

2D_Stage/tuneavideo/models/unet_mv2d_ref.py

@@ -37,9 +37,16 @@ from diffusers.models.embeddings import (
     TimestepEmbedding,
     Timesteps,
 )
-from diffusers.models.lora import LoRALinearLayer
-
-from diffusers.models.modeling_utils import ModelMixin, load_state_dict, _load_state_dict_into_model
+try:
+    from diffusers.models.lora import LoRALinearLayer
+except ImportError:
+    LoRALinearLayer = None  # LoRA not used during inference in newer diffusers
+
+from diffusers.models.modeling_utils import ModelMixin
+try:
+    from diffusers.models.modeling_utils import load_state_dict, _load_state_dict_into_model
+except ImportError:
+    from diffusers.models.model_loading_utils import load_state_dict, _load_state_dict_into_model
 from diffusers.models.unet_2d_blocks import (
     CrossAttnDownBlock2D,
     CrossAttnUpBlock2D,
@@ -50,18 +57,37 @@ from diffusers.models.unet_2d_blocks import (
 )
 from diffusers.utils import (
     CONFIG_NAME,
-    DIFFUSERS_CACHE,
     FLAX_WEIGHTS_NAME,
-    HF_HUB_OFFLINE,
     SAFETENSORS_WEIGHTS_NAME,
     WEIGHTS_NAME,
-    _add_variant,
-    _get_model_file,
     deprecate,
     is_accelerate_available,
     is_torch_version,
     logging,
 )
+try:
+    from diffusers.utils import DIFFUSERS_CACHE
+except ImportError:
+    try:
+        from huggingface_hub.constants import HF_HUB_CACHE as DIFFUSERS_CACHE
+    except ImportError:
+        import os as _os
+        DIFFUSERS_CACHE = _os.path.join(_os.path.expanduser("~"), ".cache", "huggingface", "hub")
+try:
+    from diffusers.utils import HF_HUB_OFFLINE
+except ImportError:
+    try:
+        from huggingface_hub.constants import HF_HUB_OFFLINE
+    except ImportError:
+        import os as _os
+        HF_HUB_OFFLINE = _os.environ.get("HF_HUB_OFFLINE", "0") == "1"
+try:
+    from diffusers.utils import _add_variant, _get_model_file
+except ImportError:
+    try:
+        from diffusers.utils.hub_utils import _add_variant, _get_model_file
+    except ImportError:
+        from diffusers.utils.dynamic_modules_utils import _add_variant, _get_model_file
 from diffusers import __version__
 from tuneavideo.models.unet_mv2d_blocks import (
     CrossAttnDownBlockMV2D,