From d5e56dc6e160f2dbc9946ffdb0c35c5b195855ba Mon Sep 17 00:00:00 2001
From: mayukhdeb <mayukhmainak2000@gmail.com>
Date: Fri, 15 Mar 2024 02:04:02 -0700
Subject: [PATCH] use keyword args for sanity

---
 trainer/trainer.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/trainer/trainer.py b/trainer/trainer.py
index 5cf71fb..381d9fa 100644
--- a/trainer/trainer.py
+++ b/trainer/trainer.py
@@ -79,7 +79,11 @@ class Trainer:
             text_encoder_two,
             vae,
             unet,
-        ) = load_models(args.pretrained_model, args.device, weight_dtype)
+        ) = load_models(
+            pretrained_model = args.pretrained_model, 
+            device=args.device, 
+            weight_dtype=weight_dtype
+        )
 
         # Initialize new tokens for training.
         embedding_handler = TokenEmbeddingsHandler(