Skip to content

Commit 039eef5

Browse files
committed
remove safetensors check
1 parent 00f0bd1 commit 039eef5

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

src/diffusers/models/modeling_utils.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1176,13 +1176,8 @@ def _load_pretrained_model(
11761176
error_msgs = []
11771177

11781178
# Deal with offload
1179-
is_safetensors = False
11801179
if device_map is not None and "disk" in device_map.values():
1181-
archive_file = (
1182-
resolved_archive_file[0] if isinstance(resolved_archive_file, (list, tuple)) else resolved_archive_file
1183-
)
1184-
is_safetensors = archive_file.endswith(".safetensors")
1185-
if offload_folder is None and not is_safetensors:
1180+
if offload_folder is None:
11861181
raise ValueError(
11871182
"The current `device_map` had weights offloaded to the disk. Please provide an `offload_folder`"
11881183
" for them. Alternatively, make sure you have `safetensors` installed if the model you are using"

0 commit comments

Comments
 (0)