use keyword args for sanity

2024-03-15 02:04:02 -07:00
parent 4563279eca
commit d5e56dc6e1
1 changed files with 5 additions and 1 deletions
@@ -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(