fix diffusers

genphoto/models/unet.py

@@ -13,6 +13,7 @@ from typing import List, Optional, Tuple, Union, Dict, Any
 
 from ddiffusers.configuration_utils import ConfigMixin, register_to_config
 from ddiffusers.models.attention_processor import AttentionProcessor
+from ddiffusers.utils import SAFETENSORS_WEIGHTS_NAME
 
 from ddiffusers.models.modeling_utils import ModelMixin
 from ddiffusers.utils import BaseOutput, logging
@@ -800,7 +801,7 @@ class UNet3DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixin)
             "CrossAttnUpBlock3D"
         ]
 
-        from diffusers.utils import SAFETENSORS_WEIGHTS_NAME
+
 
         model, unused_kwargs = cls.from_config(config, return_unused_kwargs=True, **unet_additional_kwargs)
         if logger is not None: