Hugging Face's logo

hugovntr
/
flux-schnell-realism

Text-to-Image
Diffusers
English
flux
lora
image-generation
Model card Files Community
flux-schnell-realism / README.md
hugovntr's picture
hugovntr
Update README.md
c4c8e89 verified
preview code
|
raw
history blame
No virus
1.33 kB
metadata
license: apache-2.0
language:
  - en
base_model: black-forest-labs/FLUX.1-schnell
pipeline_tag: text-to-image
tags:
  - flux
  - lora
  - image-generation
  - diffusers

LoRA checkpoint aiming to improve FLUX.1 [schnell] photorealism.

The weights are provided as .safetensors file, so they are suited for every use-cases, including but not limited to:

  • SD-WebUI
  • ComfyUI
  • InvokeAI
  • Forge
  • ...

Training details

This checkpoint has been trained with a fairly small dataset of 100+ synthetic images generated and upscaled using Runtime44 All images have been manually captionned without any specific trigger word.

Training was done using kohya-ss scripts on a local machine with 16Gb of VRAM and took ~1h

Usage

With 🧨Diffusers library 

from diffusers import AutoPipelineForText2Image
import torch

pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-schnell', torch_dtype=torch.bfloat16).to('cuda')
pipeline.load_lora_weights('hugovntr/flux-schnell-realism', weight_name='schnell-realism_v1')
image = pipeline('Moody kitchen at dusk, warm golden ...').images[0]
image.save("output.png")