From 3396ca881ed3f3521617cd9024eea56975191d32 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Thu, 5 Jan 2023 13:26:32 +0100
Subject: Update

---
 train_ti.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

(limited to 'train_ti.py')

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,
-        noise_scheduler=noise_scheduler,
-        unet=unet,
-        prompt_processor=prompt_processor,
-        num_class_images=args.num_class_images,
-        prior_loss_weight=args.prior_loss_weight,
-        seed=args.seed,
+        vae,
+        noise_scheduler,
+        unet,
+        prompt_processor,
+        args.num_class_images,
+        args.prior_loss_weight,
+        args.seed,
     )
 
     # We need to initialize the trackers we use, and also store our configuration.
-- 
cgit v1.2.3-54-g00ecf