up

run_lora

@@ -13,6 +13,7 @@ from io import BytesIO
 
 # path = sys.argv[1]
 path = "runwayml/stable-diffusion-v1-5"
+path = "gsdf/Counterfeit-V2.5"
 # path = "stabilityai/stable-diffusion-2-1"
 
 api = HfApi()
@@ -29,12 +30,14 @@ prompt = "masterpiece, best quality, 1girl, at dusk"
 negative_prompt = ("(low quality, worst quality:1.4), (bad anatomy), (inaccurate limb:1.2), "
                    "bad composition, inaccurate eyes, extra digit, fewer digits, (extra arms:1.2), large breasts")
 
+pipe.enable_xformers_memory_efficient_attention()
 images = pipe(prompt=prompt, 
     negative_prompt=negative_prompt, 
     width=512, 
     height=768, 
     num_inference_steps=15, 
     num_images_per_prompt=4,
+    cross_attention_kwargs={"scale": 0.5},
     generator=torch.manual_seed(0)
 ).images