prithivMLmods's picture
Update README.md
f5bc5dc verified
metadata
license: apache-2.0
language:
  - en
base_model:
  - Qwen/Qwen-Image-Edit-2511
pipeline_tag: image-to-image
library_name: diffusers
tags:
  - lora
  - black-and-white
  - true-color

1

QIE-2511-BW2TrueColor

QIE-2511-BW2TrueColor is an adapter LoRA developed for Qwen’s Qwen-Image-Edit-2511 image-to-image model, designed to transform black-and-white images into true, natural colors. The model reconstructs plausible color information while preserving the original identity, composition, textures, lighting, and geometry. It delivers clean, artifact-free results with balanced tones, controlled contrast, and consistent color realism.

Quick Start with Diffusers

Compatible with versions 2509 and 2511.

pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
from diffusers.utils import load_image

# Switch to "mps" for Apple devices
pipe = DiffusionPipeline.from_pretrained(
    "Qwen/Qwen-Image-Edit-2511",
    dtype=torch.bfloat16,
    device_map="cuda"
)

pipe.load_lora_weights("prithivMLmods/QIE-2511-BW2TrueColor")

prompt = "Transform Black and White into True Colors"

input_image = load_image(
    "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png"
)

image = pipe(image=input_image, prompt=prompt).images[0]

Trained on ModelScope.cn

Trigger Prompt

Use the following prompt to activate the black-and-white to color behavior:

Transform Black and White into True Colors

Download Model

You can download the model files from the Files & versions tab: Download