Fixed linter issues

Author: suryasidd

examples/models/stable_diffusion/model.py

@@ -173,7 +173,11 @@ def get_dummy_inputs(self):
 
         unet_inputs = (
             torch.randn(
-                batch_size, latent_channels, latent_height, latent_width, dtype=self.dtype
+                batch_size,
+                latent_channels,
+                latent_height,
+                latent_width,
+                dtype=self.dtype,
             ),
             torch.tensor([981]),  # Random timestep
             torch.randn(batch_size, text_seq_len, text_embed_dim, dtype=self.dtype),

examples/openvino/stable_diffusion/export_lcm.py

@@ -9,15 +9,16 @@
 import argparse
 import logging
 import os
-import sys
 
 import torch
-from torch.export import export
 
 from executorch.backends.openvino.partitioner import OpenvinoPartitioner
+from executorch.examples.models.stable_diffusion.model import (  # type: ignore[import-untyped]
+    LCMModelLoader,
+)
 from executorch.exir import ExecutorchBackendConfig, to_edge_transform_and_lower
 from executorch.exir.backend.backend_details import CompileSpec
-from executorch.examples.models.stable_diffusion.model import LCMModelLoader
+from torch.export import export
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -48,7 +49,9 @@ def export_text_encoder(self, output_path: str, device: str = "CPU") -> bool:
             dummy_inputs = self.model_loader.get_dummy_inputs()
 
             # Export to ATEN graph
-            exported_program = export(text_encoder_wrapper, dummy_inputs["text_encoder"])
+            exported_program = export(
+                text_encoder_wrapper, dummy_inputs["text_encoder"]
+            )
 
             # Configure OpenVINO compilation
             compile_spec = [CompileSpec("device", device.encode())]

examples/openvino/stable_diffusion/openvino_lcm.py

@@ -111,7 +111,9 @@ def encode_prompt(self, prompt: str):
             exec_time = time.time() - exec_start
             self.exec_time += exec_time
 
-            logger.info(f"Text encoder - Load: {load_time:.3f}s, Execute: {exec_time:.3f}s")
+            logger.info(
+                f"Text encoder - Load: {load_time:.3f}s, Execute: {exec_time:.3f}s"
+            )
             return embeddings
         except Exception as e:
             logger.error(f"Failed to encode prompt: {e}")
@@ -262,7 +264,7 @@ def generate_image(
         total_time = time.time() - total_start
 
         logger.info("=" * 60)
-        logger.info(f"✓ Generation completed!")
+        logger.info("✓ Generation completed!")
         logger.info(f"  Total time: {total_time:.3f}s")
         logger.info(f"  Total load time: {self.models_load_time:.3f}s")
         logger.info(f"  Total Inference time: {self.exec_time:.3f}s")