Ben-Brand-Fast-LoKr / README.md
davidrd123's picture
Update README.md
74419ef verified
metadata
license: other
base_model: black-forest-labs/FLUX.1-dev
tags:
  - flux
  - flux-diffusers
  - text-to-image
  - diffusers
  - simpletuner
  - lora
  - template:sd-lora
  - lycoris
inference: true
widget:
  - text: unconditional (blank prompt)
    parameters:
      negative_prompt: blurry, cropped, ugly
    output:
      url: ./assets/image_0_0.png
  - text: >-
      In the style of a b3nbr4nd painting, A steaming bowl of ramen with
      chopsticks resting on the edge, against a background of concentric orange
      and blue circles. The noodles are detailed in a geometric pattern and the
      steam creates a rhythmic design.
    parameters:
      negative_prompt: blurry, cropped, ugly
    output:
      url: ./assets/image_1_0.png
  - text: >-
      In the style of a b3nbr4nd painting, A vintage record player with vinyl
      spinning, set on a yellow table. The background features an alternating
      chevron pattern in purple and green. The turntable's mechanical parts are
      rendered in precise geometric shapes.
    parameters:
      negative_prompt: blurry, cropped, ugly
    output:
      url: ./assets/image_2_0.png
  - text: >-
      In the style of a b3nbr4nd painting, A sleeping cat curled up in a
      modernist chair, with a background of interlocking hexagons in red and
      blue. The cat's fur is stylized into rhythmic curves, matching the
      geometric environment.
    parameters:
      negative_prompt: blurry, cropped, ugly
    output:
      url: ./assets/image_3_0.png
  - text: >-
      In the style of a b3nbr4nd painting, A classic motorcycle viewed from the
      side, against a backdrop of radiating diamond patterns in teal and gold.
      The chrome parts reflect abstract shapes, and the wheels create circular
      motifs in the composition.
    parameters:
      negative_prompt: blurry, cropped, ugly
    output:
      url: ./assets/image_4_0.png
  - text: >-
      In the style of a b3nbr4nd painting, Portrait of a woman with silver hair
      wearing dotted blue glasses and a white lace collar, against a swirling
      background of green and yellow patterns. The background features geometric
      circles and zigzag designs.
    parameters:
      negative_prompt: blurry, cropped, ugly
    output:
      url: ./assets/image_5_0.png
  - text: >-
      In the style of a b3nbr4nd painting, A storefront sign for 'Golden Palace
      Noodles' in both English and Chinese characters, mounted on a tall pole
      against a geometric cityscape with blue and tan buildings. A small arrow
      points to available parking.
    parameters:
      negative_prompt: blurry, cropped, ugly
    output:
      url: ./assets/image_6_0.png
  - text: >-
      In the style of a b3nbr4nd painting, Dark purple figs sliced in half on a
      terra cotta plate, revealing their seeded interiors. The background
      features a repeating pattern of blue and yellow squares, with wavy lines
      creating a dynamic lower section.
    parameters:
      negative_prompt: blurry, cropped, ugly
    output:
      url: ./assets/image_7_0.png
  - text: >-
      In the style of a b3nbr4nd painting, Two young people wearing matching
      navy shirts and light gray face masks, posed against a warm yellow
      background. Their curly hair and gentle head tilts create a symmetrical
      composition.
    parameters:
      negative_prompt: blurry, cropped, ugly
    output:
      url: ./assets/image_8_0.png
  - text: >-
      In the style of a b3nbr4nd painting, A hamster wearing tiny glasses and a
      bowtie sitting at a miniature desk with a tiny laptop, against a
      background of spiral patterns in teal and orange. Office supplies scaled
      to hamster-size are arranged neatly on the desk.
    parameters:
      negative_prompt: blurry, cropped, ugly
    output:
      url: ./assets/image_9_0.png
  - text: >-
      In the style of a b3nbr4nd painting, A bearded man in a plaid shirt and
      denim apron carefully sanding a mid-century modern chair, surrounded by
      woodworking tools. The background features overlapping triangles in rust
      and navy blue colors, with sawdust creating delicate patterns in the air.
    parameters:
      negative_prompt: blurry, cropped, ugly
    output:
      url: ./assets/image_10_0.png

Ben-Brand-Fast-LoKr

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

No validation prompt was used during training.

None

Validation settings

  • CFG: 3.0
  • CFG Rescale: 0.0
  • Steps: 20
  • Sampler: None
  • Seed: 42
  • Resolution: 1024x1024

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

You can find some example images in the following gallery:

Prompt
unconditional (blank prompt)
Negative Prompt
blurry, cropped, ugly
Prompt
In the style of a b3nbr4nd painting, A steaming bowl of ramen with chopsticks resting on the edge, against a background of concentric orange and blue circles. The noodles are detailed in a geometric pattern and the steam creates a rhythmic design.
Negative Prompt
blurry, cropped, ugly
Prompt
In the style of a b3nbr4nd painting, A vintage record player with vinyl spinning, set on a yellow table. The background features an alternating chevron pattern in purple and green. The turntable's mechanical parts are rendered in precise geometric shapes.
Negative Prompt
blurry, cropped, ugly
Prompt
In the style of a b3nbr4nd painting, A sleeping cat curled up in a modernist chair, with a background of interlocking hexagons in red and blue. The cat's fur is stylized into rhythmic curves, matching the geometric environment.
Negative Prompt
blurry, cropped, ugly
Prompt
In the style of a b3nbr4nd painting, A classic motorcycle viewed from the side, against a backdrop of radiating diamond patterns in teal and gold. The chrome parts reflect abstract shapes, and the wheels create circular motifs in the composition.
Negative Prompt
blurry, cropped, ugly
Prompt
In the style of a b3nbr4nd painting, Portrait of a woman with silver hair wearing dotted blue glasses and a white lace collar, against a swirling background of green and yellow patterns. The background features geometric circles and zigzag designs.
Negative Prompt
blurry, cropped, ugly
Prompt
In the style of a b3nbr4nd painting, A storefront sign for 'Golden Palace Noodles' in both English and Chinese characters, mounted on a tall pole against a geometric cityscape with blue and tan buildings. A small arrow points to available parking.
Negative Prompt
blurry, cropped, ugly
Prompt
In the style of a b3nbr4nd painting, Dark purple figs sliced in half on a terra cotta plate, revealing their seeded interiors. The background features a repeating pattern of blue and yellow squares, with wavy lines creating a dynamic lower section.
Negative Prompt
blurry, cropped, ugly
Prompt
In the style of a b3nbr4nd painting, Two young people wearing matching navy shirts and light gray face masks, posed against a warm yellow background. Their curly hair and gentle head tilts create a symmetrical composition.
Negative Prompt
blurry, cropped, ugly
Prompt
In the style of a b3nbr4nd painting, A hamster wearing tiny glasses and a bowtie sitting at a miniature desk with a tiny laptop, against a background of spiral patterns in teal and orange. Office supplies scaled to hamster-size are arranged neatly on the desk.
Negative Prompt
blurry, cropped, ugly
Prompt
In the style of a b3nbr4nd painting, A bearded man in a plaid shirt and denim apron carefully sanding a mid-century modern chair, surrounded by woodworking tools. The background features overlapping triangles in rust and navy blue colors, with sawdust creating delicate patterns in the air.
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: 6
  • Training steps: 3600
  • Learning rate: 0.001
  • Max grad norm: 0.1
  • Effective batch size: 6
    • Micro-batch size: 3
    • Gradient accumulation steps: 2
    • Number of GPUs: 1
  • Prediction type: flow-matching (flux parameters=['shift=3', 'flux_guidance_value=1.0'])
  • Rescaled betas zero SNR: False
  • Optimizer: adamw_bf16
  • Precision: Pure BF16
  • Quantised: Yes: int8-quanto
  • Xformers: Not used
  • LyCORIS Config:
{
    "algo": "lokr",
    "multiplier": 1.0,
    "linear_dim": 10000,
    "linear_alpha": 1,
    "factor": 16,
    "apply_preset": {
        "target_module": [
            "Attention",
            "FeedForward"
        ],
        "module_algo_map": {
            "Attention": {
                "factor": 16
            },
            "FeedForward": {
                "factor": 8
            }
        }
    }
}

Datasets

benbrand-512

  • Repeats: 11
  • Total number of images: 98
  • Total number of aspect buckets: 1
  • Resolution: 0.262144 megapixels
  • Cropped: False
  • Crop style: None
  • Crop aspect: None
  • Used for regularisation data: No

benbrand-768

  • Repeats: 11
  • Total number of images: 98
  • Total number of aspect buckets: 7
  • Resolution: 0.589824 megapixels
  • Cropped: False
  • Crop style: None
  • Crop aspect: None
  • Used for regularisation data: No

benbrand-1024

  • Repeats: 5
  • Total number of images: 98
  • Total number of aspect buckets: 1
  • Resolution: 1.048576 megapixels
  • Cropped: False
  • Crop style: None
  • Crop aspect: None
  • Used for regularisation data: No

benbrand-1536

  • Repeats: 2
  • Total number of images: 98
  • Total number of aspect buckets: 2
  • Resolution: 2.359296 megapixels
  • Cropped: False
  • Crop style: None
  • Crop aspect: None
  • Used for regularisation data: No

Inference

import torch
from diffusers import DiffusionPipeline
from lycoris import create_lycoris_from_weights

model_id = 'black-forest-labs/FLUX.1-dev'
adapter_id = 'pytorch_lora_weights.safetensors' # you will have to download this manually
lora_scale = 1.0
wrapper, _ = create_lycoris_from_weights(lora_scale, adapter_id, pipeline.transformer)
wrapper.merge_to()

prompt = "An astronaut is riding a horse through the jungles of Thailand."

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")