From 0bd238a52286d66795e528b72489191cd5e96f4f Mon Sep 17 00:00:00 2001 From: Aryan Date: Sun, 15 Sep 2024 00:07:56 +0200 Subject: [PATCH] make style --- examples/cogvideo/train_cogvideox_lora.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/examples/cogvideo/train_cogvideox_lora.py b/examples/cogvideo/train_cogvideox_lora.py index 89122ce179..93b90f50b9 100644 --- a/examples/cogvideo/train_cogvideox_lora.py +++ b/examples/cogvideo/train_cogvideox_lora.py @@ -378,9 +378,7 @@ def get_args(): help="Epsilon value for the Adam optimizer and Prodigy optimizers.", ) parser.add_argument("--max_grad_norm", default=1.0, type=float, help="Max gradient norm.") - parser.add_argument( - "--prodigy_use_bias_correction", action="store_true", help="Turn on Adam's bias correction." - ) + parser.add_argument("--prodigy_use_bias_correction", action="store_true", help="Turn on Adam's bias correction.") parser.add_argument( "--prodigy_safeguard_warmup", action="store_true",