Flux_finetune_with_swami_Vivekananda
Inference
import torch
from diffusers import FluxPipeline
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
pipe.load_lora_weights("Nitin9sitare/Flux_finetune_with_swami_Vivekananda_final", weight_name="Nitin9sitare/Flux_finetune_with_swami_Vivekananda_final.safetensors")
pipe.fuse_lora(lora_scale=1.0)
pipe.to("cuda")
prompt = "A young Indian monk with sunglasses in a cricket ground, holding a bat in his hand."
image = pipe(prompt,
num_inference_steps=24,
guidance_scale=3.5,
width=512, height=512,
).images[0]
image.save(f"example.png")
Model description
This model leverages the powerful Flux architecture, a groundbreaking advancement in the world of Generative AI, specifically designed to generate personalized images. While popular tools like DALLE and MidJourney are great for generic imagery, they fall short when it comes to producing high-quality images of specific individuals. Flux revolutionizes this process by making it easy and affordable to fine-tune models with just a small set of personal images.
In this model, we demonstrate how you can fine-tune the Flux model using just 40 images, resulting in personalized image generation that captures unique facial features and characteristics with high fidelity. Whether you're looking to recreate images of yourself, historical figures, or specific individuals in creative settings, this model opens up new possibilities.
Key trigger words for this model include "A young Indian Monk", inspired by the legendary figure Swami Vivekananda, which helps direct the image generation towards that persona. The fine-tuning process is both efficient and cost-effective, running on Nvidia H100 GPUs for less than USD 2.0 (~INR 170), making this ideal for creative projects such as personalized portraits or thematic photoshoots. Model Use:
Trigger Word: "A young Indian Monk"
Fine-tuned to evoke the likeness of Swami Vivekananda.
Download the model weights in the Safetensors format from the Files & Versions tab to get started.
Trigger words
You should use A young Indian Monk
to trigger the image generation.
Download model
Weights for this model are available in Safetensors format.
Download them in the Files & versions tab.
- Downloads last month
- 60
Model tree for Nitin9sitare/Flux_finetune_with_swami_Vivekananda_final
Base model
black-forest-labs/FLUX.1-dev