1 files changed, 7 insertions, 7 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py
index 1fd86b1..4d1e0a3 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -861,13 +861,13 @@ def main():
    loop = partial(
        run_model,
-        vae=vae,
+        vae,
-        noise_scheduler=noise_scheduler,
+        noise_scheduler,
-        unet=unet,
+        unet,
-        prompt_processor=prompt_processor,
+        prompt_processor,
-        num_class_images=args.num_class_images,
+        args.num_class_images,
-        prior_loss_weight=args.prior_loss_weight,
+        args.prior_loss_weight,
-        seed=args.seed,
+        args.seed,
    )
    # We need to initialize the trackers we use, and also store our configuration.

diff --git a/train_dreambooth.py b/train_dreambooth.py index 1fd86b1..4d1e0a3 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py
@@ -861,13 +861,13 @@ def main():
861		861
862	loop = partial(	862	loop = partial(
863	run_model,	863	run_model,
864	vae=vae,	864	vae,
865	noise_scheduler=noise_scheduler,	865	noise_scheduler,
866	unet=unet,	866	unet,
867	prompt_processor=prompt_processor,	867	prompt_processor,
868	num_class_images=args.num_class_images,	868	args.num_class_images,
869	prior_loss_weight=args.prior_loss_weight,	869	args.prior_loss_weight,
870	seed=args.seed,	870	args.seed,
871	)	871	)
872		872
873	# We need to initialize the trackers we use, and also store our configuration.	873	# We need to initialize the trackers we use, and also store our configuration.