1 files changed, 7 insertions, 7 deletions
diff --git a/train_ti.py b/train_ti.py
index 164cf67..98385dd 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -814,13 +814,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_ti.py b/train_ti.py index 164cf67..98385dd 100644 --- a/train_ti.py +++ b/train_ti.py
@@ -814,13 +814,13 @@ def main():
814		814
815	loop = partial(	815	loop = partial(
816	run_model,	816	run_model,
817	vae=vae,	817	vae,
818	noise_scheduler=noise_scheduler,	818	noise_scheduler,
819	unet=unet,	819	unet,
820	prompt_processor=prompt_processor,	820	prompt_processor,
821	num_class_images=args.num_class_images,	821	args.num_class_images,
822	prior_loss_weight=args.prior_loss_weight,	822	args.prior_loss_weight,
823	seed=args.seed,	823	args.seed,
824	)	824	)
825		825
826	# We need to initialize the trackers we use, and also store our configuration.	826	# We need to initialize the trackers we use, and also store our configuration.