diff --git a/setup.py b/setup.py
index 1447f67c0d..9b7af775d5 100644
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,10 @@ def get_version() -> str:
     "pydantic>1.1,<2.0; python_version=='3.8'",
 ]
 
-extras["torch"] = ["torch", "safetensors"]
+extras["torch"] = [
+    "torch",
+    "safetensors",
+]
 extras["hf_transfer"] = [
     "hf_transfer>=0.1.4",  # Pin for progress bars
 ]