Skip to content

Commit 4127123

Browse files
committed
fixed style error
1 parent 00c29f9 commit 4127123

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/dreambooth/train_dreambooth_lora.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1122,7 +1122,7 @@ def compute_text_embeddings(prompt):
11221122
# Check the PR https://github.com/huggingface/diffusers/pull/8312 for detailed explanation.
11231123
num_warmup_steps_for_scheduler = args.lr_warmup_steps * accelerator.num_processes
11241124
if args.max_train_steps is None:
1125-
len_train_dataloader_after_sharding = ceil(len(train_dataloader) / accelerator.num_processes)
1125+
len_train_dataloader_after_sharding = math.ceil(len(train_dataloader) / accelerator.num_processes)
11261126
num_update_steps_per_epoch = math.ceil(len_train_dataloader_after_sharding / args.gradient_accumulation_steps)
11271127
num_training_steps_for_scheduler = (
11281128
args.num_train_epochs * accelerator.num_processes * num_update_steps_per_epoch

0 commit comments

Comments
 (0)