Update formatting

Author: ZachNagengast

.flake8

@@ -11,7 +11,6 @@ repos:
       - id: black
         name: black
         language: python
-        args: ["--config", ".flake8"]
 
   - repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v4.5.0

.pre-commit-config.yaml

@@ -655,7 +655,9 @@ def encode_text(
             text,
             (negative_text if cfg_weight > 1 else None),
         )
-        padded_tokens_t5 = mx.zeros((1, T5_MAX_LENGTH[self.model_version])).astype(tokens_t5.dtype)
+        padded_tokens_t5 = mx.zeros((1, T5_MAX_LENGTH[self.model_version])).astype(
+            tokens_t5.dtype
+        )
         padded_tokens_t5[:, : tokens_t5.shape[1]] = tokens_t5[
             [0], :
         ]  # Ignore negative text

python/src/diffusionkit/mlx/__init__.py

@@ -948,7 +948,7 @@ def __init__(self, in_dim: int, hidden_dim: int):
         self.mlp = nn.Sequential(
             nn.Linear(in_dim, hidden_dim),
             nn.SiLU(),
-            nn.Linear(hidden_dim, hidden_dim)
+            nn.Linear(hidden_dim, hidden_dim),
         )
 
     def __call__(self, x):

python/src/diffusionkit/mlx/mmdit.py

@@ -29,7 +29,7 @@ def run(self):
         "argmaxtools>=0.1.13",
         "torch",
         "safetensors",
-        "mlx>=0.16.3",
+        "mlx>=0.17.1",
         "jaxtyping",
         "transformers",
         "pillow",