import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("fill-in-base-model", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("dolphinium/flux_1_dev_wikiart_impressionism")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Flux Impressionism LoRA v1
This is a LoRA fine-tuning of the FLUX.1 model trained on a curated dataset of impressionist paintings from WikiArt.
Dataset
The model was trained on the WikiArt Impressionism Curated Dataset, which contains 1,000 high-quality Impressionist paintings with the following distribution:
- Landscapes: 300 images (30%)
- Portraits: 300 images (30%)
- Urban Scenes: 200 images (20%)
- Still Life: 200 images (20%)
Model Details
- Base Model: FLUX.1
- LoRA Rank: 16
- Training Steps: 2000
- Resolution: 512-1024px
Usage
from diffusers import StableDiffusionPipeline
import torch
model_id = "black-forest-labs/FLUX.1-dev"
lora_model_path = "dolphinium/flux_1_dev_wikiart_impressionism"
pipe = StableDiffusionPipeline.from_pretrained(
model_id,
torch_dtype=torch.float16
).to("cuda")
# Load LoRA weights
pipe.unet.load_attn_procs(lora_model_path)
# Generate image
prompt = "an impressionist style landscape with rolling hills and autumn trees"
image = pipe(prompt).images[0]
image.save("impressionist_landscape.png")
License
This model inherits the license of the base FLUX.1 model and the WikiArt dataset.
- Downloads last month
- 22