From 27b18776ba6d38d6bda5e5bafee3e7c4ca8c9712 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sat, 24 Jun 2023 16:26:22 +0200 Subject: Fixes --- infer.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'infer.py') 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() -- cgit v1.2.3-54-g00ecf