Skip to content

Commit 5e48cd2

Browse files
YanivDorGalronsayakpaulyiyixuxu
authored
making formatted_images initialization compact (#10801)
compact writing Co-authored-by: Sayak Paul <[email protected]> Co-authored-by: YiYi Xu <[email protected]>
1 parent 5551506 commit 5e48cd2

File tree

6 files changed

+6
-18
lines changed

6 files changed

+6
-18
lines changed

examples/controlnet/train_controlnet.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -152,9 +152,7 @@ def log_validation(
152152
validation_prompt = log["validation_prompt"]
153153
validation_image = log["validation_image"]
154154

155-
formatted_images = []
156-
157-
formatted_images.append(np.asarray(validation_image))
155+
formatted_images = [np.asarray(validation_image)]
158156

159157
for image in images:
160158
formatted_images.append(np.asarray(image))

examples/controlnet/train_controlnet_flux.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -166,9 +166,7 @@ def log_validation(
166166
validation_prompt = log["validation_prompt"]
167167
validation_image = log["validation_image"]
168168

169-
formatted_images = []
170-
171-
formatted_images.append(np.asarray(validation_image))
169+
formatted_images = [np.asarray(validation_image)]
172170

173171
for image in images:
174172
formatted_images.append(np.asarray(image))

examples/controlnet/train_controlnet_sdxl.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -157,9 +157,7 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step,
157157
validation_prompt = log["validation_prompt"]
158158
validation_image = log["validation_image"]
159159

160-
formatted_images = []
161-
162-
formatted_images.append(np.asarray(validation_image))
160+
formatted_images = [np.asarray(validation_image)]
163161

164162
for image in images:
165163
formatted_images.append(np.asarray(image))

examples/research_projects/controlnet/train_controlnet_webdataset.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -381,9 +381,7 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step)
381381
validation_prompt = log["validation_prompt"]
382382
validation_image = log["validation_image"]
383383

384-
formatted_images = []
385-
386-
formatted_images.append(np.asarray(validation_image))
384+
formatted_images = [np.asarray(validation_image)]
387385

388386
for image in images:
389387
formatted_images.append(np.asarray(image))

examples/research_projects/pixart/train_pixart_controlnet_hf.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -164,9 +164,7 @@ def log_validation(
164164
validation_prompt = log["validation_prompt"]
165165
validation_image = log["validation_image"]
166166

167-
formatted_images = []
168-
169-
formatted_images.append(np.asarray(validation_image))
167+
formatted_images = [np.asarray(validation_image)]
170168

171169
for image in images:
172170
formatted_images.append(np.asarray(image))

examples/t2i_adapter/train_t2i_adapter_sdxl.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -141,9 +141,7 @@ def log_validation(vae, unet, adapter, args, accelerator, weight_dtype, step):
141141
validation_prompt = log["validation_prompt"]
142142
validation_image = log["validation_image"]
143143

144-
formatted_images = []
145-
146-
formatted_images.append(np.asarray(validation_image))
144+
formatted_images = [np.asarray(validation_image)]
147145

148146
for image in images:
149147
formatted_images.append(np.asarray(image))

0 commit comments

Comments
 (0)