From a0263b2e5bfdd3ce09ae79e7d5cbe58a207f1f00 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 4 Apr 2023 15:18:39 +0200 Subject: [PATCH] make style --- examples/controlnet/train_controlnet.py | 2 +- examples/controlnet/train_controlnet_flax.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/controlnet/train_controlnet.py b/examples/controlnet/train_controlnet.py index 1ac3452828..ef7cc84520 100644 --- a/examples/controlnet/train_controlnet.py +++ b/examples/controlnet/train_controlnet.py @@ -106,7 +106,7 @@ def log_validation(vae, text_encoder, tokenizer, unet, controlnet, args, acceler image_logs = [] for validation_prompt, validation_image in zip(validation_prompts, validation_images): - validation_image = Image.open(validation_image).convert('RGB') + validation_image = Image.open(validation_image).convert("RGB") images = [] diff --git a/examples/controlnet/train_controlnet_flax.py b/examples/controlnet/train_controlnet_flax.py index dab6864b07..dbdf5f3b98 100644 --- a/examples/controlnet/train_controlnet_flax.py +++ b/examples/controlnet/train_controlnet_flax.py @@ -110,7 +110,7 @@ def log_validation(controlnet, controlnet_params, tokenizer, args, rng, weight_d prompt_ids = pipeline.prepare_text_inputs(prompts) prompt_ids = shard(prompt_ids) - validation_image = Image.open(validation_image).convert('RGB') + validation_image = Image.open(validation_image).convert("RGB") processed_image = pipeline.prepare_image_inputs(num_samples * [validation_image]) processed_image = shard(processed_image) images = pipeline(