linter

AGISwarm · Oct 8, 2024 · f4e4213 · f4e4213
1 parent b7144b9
commit f4e4213
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/src/AGISwarm/text2image_ms/app.py b/src/AGISwarm/text2image_ms/app.py
@@ -87,7 +87,7 @@ def send_image(websocket: WebSocket, task_id: str, image: Image.Image, **kwargs)
         )
 
     @staticmethod
-    # pylint: disable=too-many-arguments
+    # pylint: disable=too-many-arguments, too-many-positional-arguments
     def diffusion_pipeline_step_callback(
         websocket: WebSocket,
         task_id: str,

diff --git a/src/AGISwarm/text2image_ms/diffusion_pipeline.py b/src/AGISwarm/text2image_ms/diffusion_pipeline.py
@@ -76,15 +76,15 @@ def generate(
             generator = torch.Generator()
             generator.manual_seed(gen_config.seed)
         return self.pipeline(
-                prompt=gen_config.prompt,
-                negative_prompt=gen_config.negative_prompt,
-                num_inference_steps=gen_config.num_inference_steps,
-                guidance_scale=gen_config.guidance_scale,
-                generator=generator,
-                width=gen_config.width,
-                height=gen_config.height,
-                callback_on_step_end=callback_on_step_end,
-            )["images"][0]
+            prompt=gen_config.prompt,
+            negative_prompt=gen_config.negative_prompt,
+            num_inference_steps=gen_config.num_inference_steps,
+            guidance_scale=gen_config.guidance_scale,
+            generator=generator,
+            width=gen_config.width,
+            height=gen_config.height,
+            callback_on_step_end=callback_on_step_end,
+        )["images"][0]
 
     def decode_latents(self, latents: torch.Tensor, output_type: str = "pil"):
         """