format w/ black

acestep/__init__.py

@@ -4,4 +4,4 @@ ACE-Step: A Step Towards Music Generation Foundation Model
 https://github.com/ace-step/ACE-Step
 
 Apache 2.0 License
-"""
+"""

acestep/apg_guidance.py

@@ -25,7 +25,9 @@ def project(
     v1 = torch.nn.functional.normalize(v1, dim=dims)
     v0_parallel = (v0 * v1).sum(dim=dims, keepdim=True) * v1
     v0_orthogonal = v0 - v0_parallel
-    return v0_parallel.to(dtype).to(device_type), v0_orthogonal.to(dtype).to(device_type)
+    return v0_parallel.to(dtype).to(device_type), v0_orthogonal.to(dtype).to(
+        device_type
+    )
 
 
 def apg_forward(

acestep/gui.py

@@ -16,7 +16,10 @@ from acestep.data_sampler import DataSampler
 
 @click.command()
 @click.option(
-    "--checkpoint_path", type=str, default="", help="Path to the checkpoint directory. Downloads automatically if empty."
+    "--checkpoint_path",
+    type=str,
+    default="",
+    help="Path to the checkpoint directory. Downloads automatically if empty.",
 )
 @click.option(
     "--server_name",
@@ -35,7 +38,10 @@ from acestep.data_sampler import DataSampler
     help="Whether to create a public, shareable link for the Gradio app.",
 )
 @click.option(
-    "--bf16", is_flag=True, default=True, help="Whether to use bfloat16 precision. Turn off if using MPS."
+    "--bf16",
+    is_flag=True,
+    default=True,
+    help="Whether to use bfloat16 precision. Turn off if using MPS.",
 )
 @click.option(
     "--torch_compile", is_flag=True, default=False, help="Whether to use torch.compile."

acestep/language_segmentation/utils/__init__.py

@@ -1 +1 @@
-# This file intentionally left blank for Python to recognize the directory as a package.
+# This file intentionally left blank for Python to recognize the directory as a package.

infer.py

@@ -103,4 +103,4 @@ def main(checkpoint_path, bf16, torch_compile, device_id, output_path):
 
 
 if __name__ == "__main__":
-    main()
+    main()