1 files changed, 7 insertions, 7 deletions

diff --git a/infer.py b/infer.py
index 3b3b595..0a219a5 100644
--- a/infer.py
+++ b/infer.py
@@ -46,7 +46,7 @@ default_args = {
     "model": "stabilityai/stable-diffusion-2-1",
     "precision": "fp32",
     "ti_embeddings_dir": "embeddings_ti",
-    "lora_embedding": None,
+    "lora_embeddings_dir": None,
     "output_dir": "output/inference",
     "config": None,
 }
@@ -99,7 +99,7 @@ def create_args_parser():
         type=str,
     )
     parser.add_argument(
-        "--lora_embedding",
+        "--lora_embeddings_dir",
         type=str,
     )
     parser.add_argument(
@@ -341,7 +341,7 @@ def create_scheduler(config, scheduler: str, subscheduler: Optional[str] = None)
 
 
 def create_pipeline(model, dtype):
-    print("Loading Stable Diffusion pipeline...")
+    print(f"Loading Stable Diffusion pipeline: {model}...")
 
     tokenizer = MultiCLIPTokenizer.from_pretrained(
         model, subfolder="tokenizer", torch_dtype=dtype
@@ -435,11 +435,11 @@ def generate(output_dir: Path, pipeline, args):
             negative_prompt=args.negative_prompt,
             height=args.height,
             width=args.width,
+            generator=generator,
+            guidance_scale=args.guidance_scale,
             num_images_per_prompt=args.batch_size,
             num_inference_steps=args.steps,
-            guidance_scale=args.guidance_scale,
             sag_scale=args.sag_scale,
-            generator=generator,
             image=init_image,
             strength=args.image_noise,
         ).images
@@ -527,8 +527,8 @@ def main():
 
     pipeline = create_pipeline(args.model, dtype)
 
-    load_embeddings_dir(pipeline, args.ti_embeddings_dir)
-    load_lora(pipeline, args.lora_embedding)
+    # load_embeddings_dir(pipeline, args.ti_embeddings_dir)
+    # load_lora(pipeline, args.lora_embeddings_dir)
     # pipeline.unet.load_attn_procs(args.lora_embeddings_dir)
 
     cmd_parser = create_cmd_parser()