import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("LHRuig/pitdvr")
prompt = "suit"
image = pipe(prompt).images[0]pitdvr
- Prompt
- suit
Model description
pitdvr lora
Trigger words
You should use pitdvr to trigger the image generation.
Download model
Weights for this model are available in Safetensors format.
Download them in the Files & versions tab.
- Downloads last month
- 1
Model tree for LHRuig/pitdvr
Base model
black-forest-labs/FLUX.1-dev