Lint format corrections

Author: Danamir

ai_diffusion/comfy_workflow.py

@@ -499,15 +499,13 @@ def load_inpaint_model(self, model_name: str):
     def load_fooocus_inpaint(self, head: str, patch: str):
         return self.add_cached("INPAINT_LoadFooocusInpaint", 1, head=head, patch=patch)
 
-    def nunchaku_load_flux_diffusion_model(
-            self, model_path: str, cache_threshold: float
-    ):
+    def nunchaku_load_flux_diffusion_model(self, model_path: str, cache_threshold: float):
         return self.add_cached(
             "NunchakuFluxDiTLoader", 1, model_path=model_path, cache_threshold=cache_threshold
         )
 
     def nunchaku_load_qwen_diffusion_model(
-            self, model_name: str, cpu_offload: str, num_blocks_on_gpu: int, use_pin_memory: str
+        self, model_name: str, cpu_offload: str, num_blocks_on_gpu: int, use_pin_memory: str
     ):
         return self.add_cached(
             "NunchakuQwenImageDiTLoader",
@@ -602,7 +600,7 @@ def reference_latent(self, conditioning: Output, latent: Output):
         return self.add("ReferenceLatent", 1, conditioning=conditioning, latent=latent)
 
     def text_encode_qwen_image_edit(
-            self, clip: Output, vae: Output | None, image: Output, prompt: str | Output
+        self, clip: Output, vae: Output | None, image: Output, prompt: str | Output
     ):
         return self.add(
             "TextEncodeQwenImageEdit", 1, clip=clip, vae=vae, image=image, prompt=prompt

ai_diffusion/ui/server.py

@@ -201,7 +201,7 @@ def _workload_matches(self, item: PackageItem):
                 Arch.flux_k,
                 Arch.chroma,
                 Arch.qwen,
-                Arch.qwen_e
+                Arch.qwen_e,
             ]
         )