make style

Author: pcuenca

examples/consistency_distillation/train_lcm_distill_lora_sdxl_wds.py

@@ -69,9 +69,9 @@
 MAX_SEQ_LENGTH = 77
 
 # Adjust for your dataset
-WDS_JSON_WIDTH = "width"      # original_width for LAION
-WDS_JSON_HEIGHT = "height"    # original_height for LAION
-MIN_SIZE = 700                # ~960 for LAION, ideal: 1024 if the dataset contains large images
+WDS_JSON_WIDTH = "width"  # original_width for LAION
+WDS_JSON_HEIGHT = "height"  # original_height for LAION
+MIN_SIZE = 700  # ~960 for LAION, ideal: 1024 if the dataset contains large images
 
 if is_wandb_available():
     import wandb

examples/consistency_distillation/train_lcm_distill_sdxl_wds.py

@@ -68,9 +68,9 @@
 MAX_SEQ_LENGTH = 77
 
 # Adjust for your dataset
-WDS_JSON_WIDTH = "width"      # original_width for LAION
-WDS_JSON_HEIGHT = "height"    # original_height for LAION
-MIN_SIZE = 700                # ~960 for LAION, ideal: 1024 if the dataset contains large images
+WDS_JSON_WIDTH = "width"  # original_width for LAION
+WDS_JSON_HEIGHT = "height"  # original_height for LAION
+MIN_SIZE = 700  # ~960 for LAION, ideal: 1024 if the dataset contains large images
 
 if is_wandb_available():
     import wandb