Skip to content

Commit d4e7e6c

Browse files
make style
1 parent 4bbedfb commit d4e7e6c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

scripts/convert_cosmos_to_diffusers.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -909,7 +909,9 @@ def get_args():
909909
controlnet = controlnet.to(dtype=dtype)
910910

911911
if not args.save_pipeline:
912-
transformer.save_pretrained(pathlib.Path(args.output_path) / "transformer", safe_serialization=True, max_shard_size="5GB")
912+
transformer.save_pretrained(
913+
pathlib.Path(args.output_path) / "transformer", safe_serialization=True, max_shard_size="5GB"
914+
)
913915
controlnet.save_pretrained(
914916
pathlib.Path(args.output_path) / "controlnet", safe_serialization=True, max_shard_size="5GB"
915917
)

0 commit comments

Comments
 (0)