metadata

license: creativeml-openrail-m
base_model: black-forest-labs/FLUX.1-dev
tags:
  - stable-diffusion
  - stable-diffusion-diffusers
  - text-to-image
  - diffusers
  - simpletuner
  - lora
  - template:sd-lora
inference: true

flux-dreambooth-lora-r128-dev-reddit

This is a LoRA derived from black-forest-labs/FLUX.1-dev.

The main validation prompt used during training was:

a fully-clothed photograph of an adult woman, in photograph style

Validation settings

CFG: 3.0
CFG Rescale: 0.0
Steps: 20
Sampler: None
Seed: 420420420
Resolutions: 1024x1024,1280x768,960x1152

Note: The validation settings are not necessarily the same as the training settings.

Prompt
unconditional (blank prompt)

Negative Prompt
blurry, cropped, ugly

Prompt
a breathtaking naked portrait of a woman, capturing her essence with vibrant colors and expressive features

Negative Prompt
blurry, cropped, ugly

Prompt
a high-quality, detailed photograph of a woman as a sous-chef, immersed in the art of culinary creation

Negative Prompt
blurry, cropped, ugly

Prompt
a lifelike and intimate naked portrait of a woman exploding into stardust

Negative Prompt
blurry, cropped, ugly

Prompt
a cinematic, visually stunning photo of a naked man, emphasizing his dramatic and captivating presence

Negative Prompt
blurry, cropped, ugly

Prompt
an elegant and timeless portrait of a woman, exuding grace and sophistication

Negative Prompt
blurry, cropped, ugly

Prompt
a dynamic and adventurous photo of a woman, captured in an exciting, action-filled moment

Negative Prompt
blurry, cropped, ugly

Prompt
a mysterious and enigmatic portrait of a woman, shrouded in shadows and intrigue

Negative Prompt
blurry, cropped, ugly

Prompt
a vintage-style portrait of a woman, evoking the charm and nostalgia of a bygone era

Negative Prompt
blurry, cropped, ugly

Prompt
an artistic and abstract representation of a woman, blending creativity with visual storytelling

Negative Prompt
blurry, cropped, ugly

Prompt
a futuristic and cutting-edge portrayal of a woman, set against a backdrop of advanced technology

Negative Prompt
blurry, cropped, ugly

Prompt
a beautifully crafted portrait of a woman, highlighting her natural beauty and unique features

Negative Prompt
blurry, cropped, ugly

Prompt
a powerful and striking portrait of a man, capturing his strength and character

Negative Prompt
blurry, cropped, ugly

Prompt
a playful and spirited portrait of a clunky robot, capturing its metallic, shiny innocence

Negative Prompt
blurry, cropped, ugly

Prompt
a charming and vibrant portrait of a naked chewbacca full-body photo, emphasizing his hairy personality and joy

Negative Prompt
blurry, cropped, ugly

Prompt
a charming and warm photograph of a man at the beach on a sunny day

Negative Prompt
blurry, cropped, ugly

Prompt
a cute man sitting in a roller coaster car, looking like he is enjoying the day at the theme park

Negative Prompt
blurry, cropped, ugly

Prompt
a photograph of a juggalo from insane clown posse standing with his gang of thugs

Negative Prompt
blurry, cropped, ugly

Prompt
a fully-clothed photograph of an adult woman, in photograph style

Negative Prompt
blurry, cropped, ugly

The text encoder was not trained. You may reuse the base model text encoder for inference.

Training settings

Training epochs: 2
Training steps: 44500
Learning rate: 4e-05
Effective batch size: 2
- Micro-batch size: 1
- Gradient accumulation steps: 2
- Number of GPUs: 1
Prediction type: flow-matching
Rescaled betas zero SNR: False
Optimizer: AdamW, stochastic bf16
Precision: Pure BF16
Quantised: No
Xformers: Not used
LoRA Rank: 128
LoRA Alpha: 128.0
LoRA Dropout: 0.1
LoRA initialisation style: default

Datasets

normalnudes

Repeats: 0
Total number of images: 1124
Total number of aspect buckets: 3
Resolution: 1.0 megapixels
Cropped: True
Crop style: random
Crop aspect: random

nsfw-1024

Repeats: 0
Total number of images: 10816
Total number of aspect buckets: 1
Resolution: 512 px
Cropped: True
Crop style: random
Crop aspect: square

shutterstock

Repeats: 0
Total number of images: 21076
Total number of aspect buckets: 1
Resolution: 512 px
Cropped: True
Crop style: random
Crop aspect: square

julia

Repeats: 0
Total number of images: 34
Total number of aspect buckets: 1
Resolution: 512 px
Cropped: True
Crop style: random
Crop aspect: square

riverphoenix

Repeats: 0
Total number of images: 12
Total number of aspect buckets: 1
Resolution: 512 px
Cropped: True
Crop style: random
Crop aspect: square

Inference

import torch
from diffusers import DiffusionPipeline

model_id = 'black-forest-labs/FLUX.1-dev'
adapter_id = 'ptx0/flux-dreambooth-lora-r128-dev-reddit'
pipeline = DiffusionPipeline.from_pretrained(model_id)
pipeline.load_lora_weights(adapter_id)

prompt = "a fully-clothed photograph of an adult woman, in photograph style"


pipeline.to('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu')
image = pipeline(
    prompt=prompt,
    num_inference_steps=20,
    generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(1641421826),
    width=1024,
    height=1024,
    guidance_scale=3.0,
).images[0]
image.save("output.png", format="PNG")