|
50 | 50 | if is_accelerate_available():
|
51 | 51 | from accelerate import init_empty_weights
|
52 | 52 |
|
| 53 | + from ..models.modeling_utils import load_model_dict_into_meta |
| 54 | + |
53 | 55 | logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|
54 | 56 |
|
55 | 57 | CONFIG_URLS = {
|
@@ -977,8 +979,6 @@ def create_diffusers_controlnet_model_from_ldm(
|
977 | 979 | controlnet = ControlNetModel(**diffusers_config)
|
978 | 980 |
|
979 | 981 | if is_accelerate_available():
|
980 |
| - from ..models.modeling_utils import load_model_dict_into_meta |
981 |
| - |
982 | 982 | unexpected_keys = load_model_dict_into_meta(
|
983 | 983 | controlnet, diffusers_format_controlnet_checkpoint, dtype=torch_dtype
|
984 | 984 | )
|
@@ -1155,8 +1155,6 @@ def create_text_encoder_from_ldm_clip_checkpoint(config_name, checkpoint, local_
|
1155 | 1155 | text_model_dict[diffusers_key] = checkpoint[key]
|
1156 | 1156 |
|
1157 | 1157 | if is_accelerate_available():
|
1158 |
| - from ..models.modeling_utils import load_model_dict_into_meta |
1159 |
| - |
1160 | 1158 | unexpected_keys = load_model_dict_into_meta(text_model, text_model_dict, dtype=torch_dtype)
|
1161 | 1159 | if text_model._keys_to_ignore_on_load_unexpected is not None:
|
1162 | 1160 | for pat in text_model._keys_to_ignore_on_load_unexpected:
|
@@ -1250,8 +1248,6 @@ def create_text_encoder_from_open_clip_checkpoint(
|
1250 | 1248 | text_model_dict[diffusers_key] = checkpoint[key]
|
1251 | 1249 |
|
1252 | 1250 | if is_accelerate_available():
|
1253 |
| - from ..models.modeling_utils import load_model_dict_into_meta |
1254 |
| - |
1255 | 1251 | unexpected_keys = load_model_dict_into_meta(text_model, text_model_dict, dtype=torch_dtype)
|
1256 | 1252 | if text_model._keys_to_ignore_on_load_unexpected is not None:
|
1257 | 1253 | for pat in text_model._keys_to_ignore_on_load_unexpected:
|
@@ -1317,8 +1313,6 @@ def create_diffusers_unet_model_from_ldm(
|
1317 | 1313 | unet = UNet2DConditionModel(**unet_config)
|
1318 | 1314 |
|
1319 | 1315 | if is_accelerate_available():
|
1320 |
| - from ..models.modeling_utils import load_model_dict_into_meta |
1321 |
| - |
1322 | 1316 | unexpected_keys = load_model_dict_into_meta(unet, diffusers_format_unet_checkpoint, dtype=torch_dtype)
|
1323 | 1317 | if unet._keys_to_ignore_on_load_unexpected is not None:
|
1324 | 1318 | for pat in unet._keys_to_ignore_on_load_unexpected:
|
@@ -1379,8 +1373,6 @@ def create_diffusers_vae_model_from_ldm(
|
1379 | 1373 | vae = AutoencoderKL(**vae_config)
|
1380 | 1374 |
|
1381 | 1375 | if is_accelerate_available():
|
1382 |
| - from ..models.modeling_utils import load_model_dict_into_meta |
1383 |
| - |
1384 | 1376 | unexpected_keys = load_model_dict_into_meta(vae, diffusers_format_vae_checkpoint, dtype=torch_dtype)
|
1385 | 1377 | if vae._keys_to_ignore_on_load_unexpected is not None:
|
1386 | 1378 | for pat in vae._keys_to_ignore_on_load_unexpected:
|
|
0 commit comments