You can deploy this model and start earning money today!

shibu12/
flux-lora-wheels

Name: flux-lora-wheels
Availability: InStock

black-forest-labs/FLUX.1-dev

adapter

other

Views

No views yet

Flux DreamBooth LoRA - shibu12/flux-lora-wheels

Model description

These are shibu12/flux-lora-wheels DreamBooth LoRA weights for black-forest-labs/FLUX.1-dev.

The weights were trained using DreamBooth with the Flux diffusers trainer.

Was LoRA for the text encoder enabled? False.

Pivotal tuning was enabled: True.

Trigger words

To trigger image generation of trained concept(or concepts) replace each concept identifier in you prompt with the new inserted tokens:

to trigger concept `TOK` → use `<s0><s1>` in your prompt

Download model

Download the *.safetensors LoRA in the Files & versions tab.

Use it with the 🧨 diffusers library

from diffusers import AutoPipelineForText2Image
import torch
from huggingface_hub import hf_hub_download
    from safetensors.torch import load_file
            
pipeline = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to('cuda')
pipeline.load_lora_weights('shibu12/flux-lora-wheels', weight_name='pytorch_lora_weights.safetensors')
embedding_path = hf_hub_download(repo_id='shibu12/flux-lora-wheels', filename='flux-lora-wheels_emb.safetensors', repo_type="model")
    state_dict = load_file(embedding_path)
    pipeline.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer)
    pipeline.load_textual_inversion(state_dict["t5"], token=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder_2, tokenizer=pipeline.tokenizer_2)
            
image = pipeline('a <s0><s1> wheel').images[0]

For more details, including weighting, merging and fusing LoRAs, check the documentation on loading LoRAs in diffusers

License

Please adhere to the licensing terms as described here.

Intended uses & limitations

How to use

# TODO: add an example code snippet for running this diffusion pipeline

Limitations and bias

[TODO: provide examples of latent issues and potential remediations]

Training details

[TODO: describe the data used to train the model]