|
--- |
|
tags: |
|
- text-to-image |
|
- lora |
|
- diffusers |
|
- template:diffusion-lora |
|
base_model: black-forest-labs/FLUX.1-dev |
|
instance_prompt: lehenga |
|
license: other |
|
license_name: flux-1-dev-non-commercial-license |
|
license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md |
|
language: |
|
- en |
|
widget: |
|
- text: >- |
|
A flat-lay image of A lehenga with a traditional style and a fitted |
|
waistline is elegantly crafted from stretchy silk material, ensuring a |
|
comfortable and flattering fit. The long hemline adds a touch of grace and |
|
sophistication to the ensemble. Adorned in a solid blue color, it features a |
|
sleeveless design that complements its sweetheart neckline. The solid |
|
pattern and the luxurious silk fabric together create a timeless and chic |
|
look that is perfect for special occasions. |
|
output: |
|
url: images/example_w40syzp04.png |
|
- text: >- |
|
A lehenga with a vibrant yellow color and an embroidered pattern showcases a |
|
traditional style. It features a fitted silhouette that gracefully flows |
|
with its long hemline. Crafted from a luxurious silk material, this lehenga |
|
offers a stretchable fabric elasticity for comfort and movement. The full |
|
sleeve length adds elegance, while the flared waistline accentuates the |
|
wearer's figure. The V-neck design further complements the overall |
|
aesthetic, making it a stunning choice for special occasions. |
|
output: |
|
url: images/example_bjilw3a2b.png |
|
- text: >- |
|
A lehenga with a traditional style and a fitted fit features a long hemline |
|
that elegantly drapes. The fabric is a comfortable cotton blend that |
|
includes stretchable properties, ensuring a flattering fit. Adorned with a |
|
full sleeve length, it adds a touch of grace to the ensemble. The lehenga |
|
boasts a beautiful floral pattern, with white as the dominant color, and a |
|
V-neckline that adds a subtle yet sophisticated detail. The high waistline |
|
accentuates the figure, making it a chic choice for any occasion. |
|
output: |
|
url: images/example_infawkf33.png |
|
|
|
--- |
|
# FLUX.1-dev LoRA Lehenga Generator |
|
|
|
<Gallery /> |
|
|
|
## Model description |
|
|
|
FLUX.1-dev LoRA Lehenga Generator can create an image of Lehenga by detailing the color, pattern, fit, style, material, and type. |
|
|
|
## Repository used |
|
|
|
AI Toolkit by Ostris |
|
https://github.com/ostris/ai-toolkit |
|
|
|
|
|
## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers) |
|
|
|
```py |
|
import random |
|
from diffusers import FluxPipeline |
|
import torch |
|
|
|
MAX_SEED=42 |
|
PRE_TRAINED_MODEL = "black-forest-labs/FLUX.1-dev" |
|
FINE_TUNED_MODEL = "tryonlabs/FLUX.1-dev-LoRA-Lehenga-Generator" |
|
|
|
# Load Flux.1-dev |
|
pipe = FluxPipeline.from_pretrained(PRE_TRAINED_MODEL, torch_dtype=torch.float16).to("cuda") |
|
|
|
# Load fine-tuned model |
|
pipe.load_lora_weights(FINE_TUNED_MODEL, adapter_name="default", weight_name="lehenga-generator.safetensors") |
|
|
|
seed = random.randint(0, MAX_SEED) |
|
generator = torch.Generator().manual_seed(seed) |
|
prompt = "A flat lay image of a lehenga with a pink solid pattern and traditional style is elegantly fitted, showcasing a floor-length hemline that flows beautifully. Crafted from luxurious silk, this garment features a stretchy fabric that ensures comfort and a flattering fit. The lehenga is sleeveless, allowing for unrestricted movement, and it is adorned with a sweetheart neckline that adds a touch of femininity. The fitted waistline accentuates the wearer's silhouette, making it a stunning choice for special occasions." |
|
image = pipe(prompt, height=1024, width=1024, num_images_per_prompt=1, generator=generator, guidance_scale=4.5, num_inference_steps=40).images[0] |
|
|
|
image.save("gen_image.jpg") |
|
``` |
|
|
|
## Download model |
|
|
|
Weights for this model are available in Safetensors format. |
|
|
|
[Download](/tryonlabs/FLUX.1-dev-LoRA-Outfit-Generator/tree/main) them in the Files & versions tab. |
|
|