This is a fine-tuned LoRA model based on the Flux Schnell model. It is designed for generating text-to-image outputs with specific customizations.
1from diffusers import AutoPipelineForText2Image
2import torch
3
4# Load base model
5pipeline = AutoPipelineForText2Image.from_pretrained(
6 "black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16
7).to("cuda")
8
9# Load LoRA weights
10pipeline.load_lora_weights("blackpillCO/my_first_flux_lora_v1_for_blackpill")
11
12# Generate an image
13image = pipeline("jordanbarett posing as a model").images[0]
14image.save("generated_image.png")
