From 1ade42f72998ec47147cb35e10f5d2283737e420 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 23 Oct 2023 19:43:54 +0200 Subject: [PATCH] make style --- examples/dreambooth/train_dreambooth_lora_sdxl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/dreambooth/train_dreambooth_lora_sdxl.py b/examples/dreambooth/train_dreambooth_lora_sdxl.py index 9a7b54a69c..d7df6d4ef5 100644 --- a/examples/dreambooth/train_dreambooth_lora_sdxl.py +++ b/examples/dreambooth/train_dreambooth_lora_sdxl.py @@ -1070,11 +1070,11 @@ def main(args): if args.train_text_encoder: text_encoder_one.train() text_encoder_two.train() - + # set top parameter requires_grad = True for gradient checkpointing works text_encoder_one.text_model.embeddings.requires_grad_(True) text_encoder_two.text_model.embeddings.requires_grad_(True) - + for step, batch in enumerate(train_dataloader): with accelerator.accumulate(unet): pixel_values = batch["pixel_values"].to(dtype=vae.dtype)