From f84d0910fb702580f9e10b4b78698fe3078b6a83 Mon Sep 17 00:00:00 2001 From: YanivDorGalron <89192632+YanivDorGalron@users.noreply.github.com> Date: Sun, 16 Feb 2025 15:30:27 +0200 Subject: [PATCH] compact writing --- examples/controlnet/train_controlnet.py | 4 +--- examples/controlnet/train_controlnet_flux.py | 4 +--- examples/controlnet/train_controlnet_sdxl.py | 4 +--- .../controlnet/train_controlnet_webdataset.py | 4 +--- .../research_projects/pixart/train_pixart_controlnet_hf.py | 4 +--- examples/t2i_adapter/train_t2i_adapter_sdxl.py | 4 +--- 6 files changed, 6 insertions(+), 18 deletions(-) diff --git a/examples/controlnet/train_controlnet.py b/examples/controlnet/train_controlnet.py index 9c41315ba064..343ccca45ac4 100644 --- a/examples/controlnet/train_controlnet.py +++ b/examples/controlnet/train_controlnet.py @@ -152,9 +152,7 @@ def log_validation( validation_prompt = log["validation_prompt"] validation_image = log["validation_image"] - formatted_images = [] - - formatted_images.append(np.asarray(validation_image)) + formatted_images = [np.asarray(validation_image)] for image in images: formatted_images.append(np.asarray(image)) diff --git a/examples/controlnet/train_controlnet_flux.py b/examples/controlnet/train_controlnet_flux.py index 7f93477fc5b7..a41615c7b546 100644 --- a/examples/controlnet/train_controlnet_flux.py +++ b/examples/controlnet/train_controlnet_flux.py @@ -166,9 +166,7 @@ def log_validation( validation_prompt = log["validation_prompt"] validation_image = log["validation_image"] - formatted_images = [] - - formatted_images.append(np.asarray(validation_image)) + formatted_images = [np.asarray(validation_image)] for image in images: formatted_images.append(np.asarray(image)) diff --git a/examples/controlnet/train_controlnet_sdxl.py b/examples/controlnet/train_controlnet_sdxl.py index b2d950e09ac1..17f313752989 100644 --- a/examples/controlnet/train_controlnet_sdxl.py +++ b/examples/controlnet/train_controlnet_sdxl.py @@ -157,9 +157,7 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step, validation_prompt = log["validation_prompt"] validation_image = log["validation_image"] - formatted_images = [] - - formatted_images.append(np.asarray(validation_image)) + formatted_images = [np.asarray(validation_image)] for image in images: formatted_images.append(np.asarray(image)) diff --git a/examples/research_projects/controlnet/train_controlnet_webdataset.py b/examples/research_projects/controlnet/train_controlnet_webdataset.py index 765bb495062e..829b0031156e 100644 --- a/examples/research_projects/controlnet/train_controlnet_webdataset.py +++ b/examples/research_projects/controlnet/train_controlnet_webdataset.py @@ -381,9 +381,7 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step) validation_prompt = log["validation_prompt"] validation_image = log["validation_image"] - formatted_images = [] - - formatted_images.append(np.asarray(validation_image)) + formatted_images = [np.asarray(validation_image)] for image in images: formatted_images.append(np.asarray(image)) diff --git a/examples/research_projects/pixart/train_pixart_controlnet_hf.py b/examples/research_projects/pixart/train_pixart_controlnet_hf.py index 995a20dfa28e..67ec30da0ece 100644 --- a/examples/research_projects/pixart/train_pixart_controlnet_hf.py +++ b/examples/research_projects/pixart/train_pixart_controlnet_hf.py @@ -164,9 +164,7 @@ def log_validation( validation_prompt = log["validation_prompt"] validation_image = log["validation_image"] - formatted_images = [] - - formatted_images.append(np.asarray(validation_image)) + formatted_images = [np.asarray(validation_image)] for image in images: formatted_images.append(np.asarray(image)) diff --git a/examples/t2i_adapter/train_t2i_adapter_sdxl.py b/examples/t2i_adapter/train_t2i_adapter_sdxl.py index 935d53a48b34..a34ecf17eb30 100644 --- a/examples/t2i_adapter/train_t2i_adapter_sdxl.py +++ b/examples/t2i_adapter/train_t2i_adapter_sdxl.py @@ -141,9 +141,7 @@ def log_validation(vae, unet, adapter, args, accelerator, weight_dtype, step): validation_prompt = log["validation_prompt"] validation_image = log["validation_image"] - formatted_images = [] - - formatted_images.append(np.asarray(validation_image)) + formatted_images = [np.asarray(validation_image)] for image in images: formatted_images.append(np.asarray(image))