Skip to content

Commit 0cbc78f

Browse files
authored
[Modeling utils chore] import load_model_dict_into_meta only once (huggingface#7437)
import load_model_dict_into_meta only once
1 parent 0cc5630 commit 0cbc78f

File tree

1 file changed

+2
-10
lines changed

1 file changed

+2
-10
lines changed

src/diffusers/loaders/single_file_utils.py

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,8 @@
5050
if is_accelerate_available():
5151
from accelerate import init_empty_weights
5252

53+
from ..models.modeling_utils import load_model_dict_into_meta
54+
5355
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
5456

5557
CONFIG_URLS = {
@@ -977,8 +979,6 @@ def create_diffusers_controlnet_model_from_ldm(
977979
controlnet = ControlNetModel(**diffusers_config)
978980

979981
if is_accelerate_available():
980-
from ..models.modeling_utils import load_model_dict_into_meta
981-
982982
unexpected_keys = load_model_dict_into_meta(
983983
controlnet, diffusers_format_controlnet_checkpoint, dtype=torch_dtype
984984
)
@@ -1155,8 +1155,6 @@ def create_text_encoder_from_ldm_clip_checkpoint(config_name, checkpoint, local_
11551155
text_model_dict[diffusers_key] = checkpoint[key]
11561156

11571157
if is_accelerate_available():
1158-
from ..models.modeling_utils import load_model_dict_into_meta
1159-
11601158
unexpected_keys = load_model_dict_into_meta(text_model, text_model_dict, dtype=torch_dtype)
11611159
if text_model._keys_to_ignore_on_load_unexpected is not None:
11621160
for pat in text_model._keys_to_ignore_on_load_unexpected:
@@ -1250,8 +1248,6 @@ def create_text_encoder_from_open_clip_checkpoint(
12501248
text_model_dict[diffusers_key] = checkpoint[key]
12511249

12521250
if is_accelerate_available():
1253-
from ..models.modeling_utils import load_model_dict_into_meta
1254-
12551251
unexpected_keys = load_model_dict_into_meta(text_model, text_model_dict, dtype=torch_dtype)
12561252
if text_model._keys_to_ignore_on_load_unexpected is not None:
12571253
for pat in text_model._keys_to_ignore_on_load_unexpected:
@@ -1317,8 +1313,6 @@ def create_diffusers_unet_model_from_ldm(
13171313
unet = UNet2DConditionModel(**unet_config)
13181314

13191315
if is_accelerate_available():
1320-
from ..models.modeling_utils import load_model_dict_into_meta
1321-
13221316
unexpected_keys = load_model_dict_into_meta(unet, diffusers_format_unet_checkpoint, dtype=torch_dtype)
13231317
if unet._keys_to_ignore_on_load_unexpected is not None:
13241318
for pat in unet._keys_to_ignore_on_load_unexpected:
@@ -1379,8 +1373,6 @@ def create_diffusers_vae_model_from_ldm(
13791373
vae = AutoencoderKL(**vae_config)
13801374

13811375
if is_accelerate_available():
1382-
from ..models.modeling_utils import load_model_dict_into_meta
1383-
13841376
unexpected_keys = load_model_dict_into_meta(vae, diffusers_format_vae_checkpoint, dtype=torch_dtype)
13851377
if vae._keys_to_ignore_on_load_unexpected is not None:
13861378
for pat in vae._keys_to_ignore_on_load_unexpected:

0 commit comments

Comments
 (0)