RedbeardNZ/LBM_relighting

✨ Latent Bridge Matching for Image Relighting ✨

Latent Bridge Matching (LBM) is a new, versatile and scalable method proposed in LBM: Latent Bridge Matching for Fast Image-to-Image Translation that relies on Bridge Matching in a latent space to achieve fast image-to-image translation. This model was trained to relight a foreground object according to a provided background. See our live demo and official Github repo.

How to use?

To use this model you need first to install the associated lbm library by running the following

pip install git+https://github.com/gojasper/LBM.git

Then, you can infer with the model on your input images

import torch
from diffusers.utils import load_image
from lbm.inference import evaluate, get_model

# Load model
model = get_model(
  "jasperai/LBM_relighting",
  torch_dtype=torch.bfloat16,
  device="cuda",
)

# Load a source image
source_image = load_image(
  "https://huggingface.co/jasperai/LBM_relighting/resolve/main/assets/source_image.jpg"
)

# Perform inference
output_image = evaluate(model, source_image, num_sampling_steps=1)
output_image

License

This code is released under the Creative Commons BY-NC 4.0 license.

Citation

If you find this work useful or use it in your research, please consider citing us

@article{chadebec2025lbm,
      title={LBM: Latent Bridge Matching for Fast Image-to-Image Translation}, 
      author={Clément Chadebec and Onur Tasar and Sanjeev Sreetharan and Benjamin Aubin},
      year={2025},
      journal = {arXiv preprint arXiv:2503.07535},
}