Skip to content

Commit e4d1553

Browse files
committed
ruff
1 parent afd115b commit e4d1553

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/diffusers/models/modeling_utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1318,7 +1318,9 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
13181318
else:
13191319
flashpack_device = device_map[""]
13201320
if flashpack_device in ["auto", "balanced", "balanced_low_0", "sequential"]:
1321-
raise ValueError("FlashPack `device_map` should be a device, not one of `auto`, `balanced`, `balanced_low_0`, `sequential`.")
1321+
raise ValueError(
1322+
"FlashPack `device_map` should be a device, not one of `auto`, `balanced`, `balanced_low_0`, `sequential`."
1323+
)
13221324

13231325
flashpack.mixin.assign_from_file(
13241326
model=model,
@@ -1342,9 +1344,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
13421344
)
13431345

13441346
if output_loading_info:
1345-
logger.warning(
1346-
"`output_loading_info` is not supported with FlashPack."
1347-
)
1347+
logger.warning("`output_loading_info` is not supported with FlashPack.")
13481348
return model, {}
13491349

13501350
return model

0 commit comments

Comments
 (0)