fix lint (#1720)

nod-ai · Aug 4, 2023 · a57eccc · a57eccc
1 parent a686d7d
commit a57eccc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/apps/stable_diffusion/scripts/train_lora_word.py b/apps/stable_diffusion/scripts/train_lora_word.py
@@ -136,7 +136,10 @@ def __getitem__(self, i):
 
         if self.center_crop:
             crop = min(img.shape[0], img.shape[1])
-            (h, w,) = (
+            (
+                h,
+                w,
+            ) = (
                 img.shape[0],
                 img.shape[1],
             )

diff --git a/apps/stable_diffusion/src/utils/utils.py b/apps/stable_diffusion/src/utils/utils.py
@@ -144,7 +144,10 @@ def compile_through_fx(
     if "cuda" in args.device:
         shark_args.enable_tf32 = True
 
-    (mlir_module, func_name,) = import_with_fx(
+    (
+        mlir_module,
+        func_name,
+    ) = import_with_fx(
         model=model,
         inputs=inputs,
         is_f16=is_f16,