Skip to content

GitLab

Commit a0263b2e authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent 62c01d26
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/controlnet/train_controlnet.py
View file @ a0263b2e
...@@ -106,7 +106,7 @@ def log_validation(vae, text_encoder, tokenizer, unet, controlnet, args, acceler ...@@ -106,7 +106,7 @@ def log_validation(vae, text_encoder, tokenizer, unet, controlnet, args, acceler
image_logs = [] image_logs = []
for validation_prompt, validation_image in zip(validation_prompts, validation_images): 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 = [] images = []
......
examples/controlnet/train_controlnet_flax.py
View file @ a0263b2e
...@@ -110,7 +110,7 @@ def log_validation(controlnet, controlnet_params, tokenizer, args, rng, weight_d ...@@ -110,7 +110,7 @@ def log_validation(controlnet, controlnet_params, tokenizer, args, rng, weight_d
prompt_ids = pipeline.prepare_text_inputs(prompts) prompt_ids = pipeline.prepare_text_inputs(prompts)
prompt_ids = shard(prompt_ids) 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 = pipeline.prepare_image_inputs(num_samples * [validation_image])
processed_image = shard(processed_image) processed_image = shard(processed_image)
images = pipeline( images = pipeline(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment