Views
No views yet
1from diffusers import StableDiffusionPipeline
2import torch
3
4model_id = "black-forest-labs/FLUX.1-dev"
5lora_model_path = "dolphinium/flux_1_dev_wikiart_impressionism"
6
7pipe = StableDiffusionPipeline.from_pretrained(
8 model_id,
9 torch_dtype=torch.float16
10).to("cuda")
11
12# Load LoRA weights
13pipe.unet.load_attn_procs(lora_model_path)
14
15# Generate image
16prompt = "an impressionist style landscape with rolling hills and autumn trees"
17image = pipe(prompt).images[0]
18image.save("impressionist_landscape.png")