Fine-tune Flux Image Models on Flux Labs AI

I am incredibly proud to launch Flux Labs!

After weeks of planning, coding, and refining the product, Flux Labs is now available to the world. The primary goal? To become the leading platform for AI image and video generation using Flux models from Black Forest Labs.

Many would ask: Why another image generator?

I know… I know. There are already tons of AI image generators out there. The market is saturated with all sorts of tools, from small startups to big names. But what sets Flux Labs apart is its use of Flux image models.

In case you didn’t know it yet, these 12 billion parameter models are state-of-the-art (SOTA) and open-weights. It means you can download them, run them locally on your machine, and generate images without needing an internet connection.

Here’s an interesting fact: Black Forest Labs is comprised of the original developers of the technology that powers Stable Diffusion and the inventors of latent diffusion. Following some internal issues at Stability AI, key team members left to form a new startup.

When I first tested Flux, I was completely blown away. The image quality, how well it adhered to prompts, and its accuracy in rendering text were simply astonishing. I’ve used MidJourney and many others, but Flux surpasses them all.

Additionally, the developers are planning to roll out a video model soon. My goal is to let the users train the model with their custom product image and turn them into video ads. The potential applications are limitless, and I’m thrilled to have launched Flux Labs now to be ahead of the curve.

Flux Labs Tech Stack

For the developers reading this, I know you’re curious about the tech under the hood, so here’s the full stack I used:

• Next.js 14 framework, TypeScript
• NextAuth for authentication
• Tailwind CSS, Framer Motion, and React for the frontend
• Vercel Postgres and Prisma for database management
• AWS S3 for image storage
• Stripe for handling subscriptions and payments
• Sanity for blog management
• Vercel for hosting and deployment

I purchased the domain via GoDaddy and use Google Search Console (GSC) and Google Analytics for SEO tracking and user insights.

Flux Labs Features

At launch, Flux Labs will offer three powerful features:

1. Image generation with Flux models : Test out three models — Schnell, Dev, and Pro — all with varying capabilities in terms of speed and quality.
2. Image model training: You can fine-tune Flux models with your own photos to create unlimited personalized images — whether that’s for yourself, pets, or products.
3. Image upscaler : This feature will allow users to upscale and enhance images by up to 4x their original size. It’s not just a simple resolution boost; there’s sophisticated AI working behind the scenes to intelligently add details and sharpness.

These features can be accessed through the AI Tools drop-down menu.

Let’s dive into how they work.

How to Generate Images

After logging in with your Google account, navigate to Image Generator under the AI Tools menu.

The image generator dashboard is simple and intuitive. You simply describe what you want in the prompt field, adjust parameters, and watch the magic happen.

Some adjustable parameters include:

• Flux model: Flux Schnell is the fastest with a good quality result; Flux Dev is slower but produces a better image; Flux Pro is the best model but takes the longest time to produce the result.
• Aspect ratio: Adjust the resolution of the image as square (1:1), portrait (4:5), landscape (16:9), etc.
• Steps: This is an integer that defines the number of steps in the image generation process. More steps generally lead to higher quality images but also increase the computation time.
• Guidance: This is also known as the Classifier-Free Guidance (CFG) scale, is a parameter that controls how closely an AI image generation process follows a text prompt.
• Seed: This number is generated randomly for each image but can be specified with the seed parameter. If you use the same seed number and prompt, you will get similar final images.
• Output format: You can choose the file format of the output image between JPG, PNG, and Webp.
• Output quality: Adjust the quality of your image between 0 and 100. The higher the number means more processing time.

If you’re unsure about settings, the default works just fine. Here’s an example:

Prompt: An image of a cute dog lying on a fluffy circle pillow, smiling for a photoshoot

Awesome! The generation speed is fast even with the Flux Pro setting and the result looks absolutely impressive. Here’s an expanded version of the result:

You can also view the images under the creations page.

How to Train Image Models

To train a custom model, click Train Model from the homepage or AI Tools menu.

On the training page, you’ll fill out the form with your model name, trigger word, and upload 4+ images.

What’s a trigger word?

The trigger_word refers to the object, style, or concept you are training on. Pick a string that isn’t a real word, like TOK, or something related to what’s being trained, like CYBRPNK. The trigger word you specify will be associated with all images during training. Then, when you run your fine-tuned model, you can include the trigger word in prompts to activate your concept.

Note: Upload 4–10 high-quality samples, front facing, square aspect ratio, 1 person in frame, various angles.

I won’t go through the step-by-step process in this article. If you want a detailed guide, you can check out my article on How to Train Flux AI Models for a step-by-step breakdown.

How To Train Flux AI Image Model To Generate Unlimited Photos of Yourself
Today, you can fine-tune the most capable open-weight image model, Flux, from Black Forest Labs to generate unlimited…jimclydemonge.medium.com

After the training is complete, you’ll see a “Use Model” button to start creating custom images.

The image generation page for fine-tuned models is somewhat the same as the previous image generation page.

In the prompt field, make sure to add the trigger word in your prompt.

Here’s an example:

Prompt: jimclyde wearing a tuxedo in a red background

I couldn’t stop laughing when I saw the results. It was like looking into a mirror — AI me in various tuxedo styles!

Image Upscaler Feature (Coming soon)

I am still working on this feature, but the plan is to use the Clarity Upscaler for this. This is by far the best upscaler model accessible via an API.

Check out the before and after example below:

The results I get are pretty close to Magnific AI.

How Much Does It Cost?

Flux Labs offers a tiered subscription model for its AI image generation and image model training platform, with three options: Pro, Premium, and Enterprise.

The Pro tier costs $9.99 per month

• 200 Flux Schnell images
• 100 Flux Dev images
• 66 Flux Pro images
• 1 custom model

The Premium tier is priced at $19.99 per month

• 500 Flux Schnell images
• 250 Flux Dev images
• 125 Flux Pro images
• 3 custom models

The Enterprise tier costs $49.99 monthly

• 1,500 Flux Schnell images
• 750 Flux Dev images
• 375 Flux Pro images
• 5 custom models

What’s Next?

There’s so much more in the pipeline for Flux Labs. Some of the features I’m currently working on include:

• Flux LoRA models for more fine-tuned generation.
• Text-to-video and image-to-video capabilities.
• The ability to generate videos based on the images you’ve already created.
• A community page for sharing prompts, fine-tuned models, and more.

I also plan on rolling out these improvements by the end of the year:

• An affiliate program.
• More pricing options, like one-time purchases or annual subscriptions.
• Enhancing the user interface and user experience.
• A marketing strategy to boost visibility on social media and improve SEO.
• Website optimizations like upgrading storage, hosting, and more AI integrations.

Final Thoughts

This is the first product I’ve built and launched entirely on my own, and I couldn’t be prouder. Of course, there are bound to be glitches and bugs, but I’m eager to address them as they arise.

My vision for Flux Labs AI is to create the go-to platform for generating custom images and fine-tuning models. I’m also passionate about empowering businesses and individuals to quickly generate high-quality, personalized images for branding purposes.

Please visit the Flux Labs AI site, try it out, and let me know your thoughts. I’d be honored if you subscribed to a plan and explored the fine-tuning feature!

‍