Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing

  • Log In
  • Sign Up

yeates
/
OmniPaint

Image-to-Image
Diffusers
image-inpainting
object-removal
object-insertion
flux
lora
iccv2025
Model card Files Files and versions
xet
Community

Instructions to use yeates/OmniPaint with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Diffusers

    How to use yeates/OmniPaint with Diffusers:

    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("black-forest-labs/FLUX.1-dev", dtype=torch.bfloat16, device_map="cuda")
    pipe.load_lora_weights("yeates/OmniPaint")
    
    prompt = "Turn this cat into a dog"
    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]
  • Notebooks
  • Google Colab
  • Kaggle
  • Local Apps
  • Draw Things
OmniPaint / weights
Ctrl+K
Ctrl+K
  • 1 contributor
History: 1 commit
yeates's picture
yeates
Upload 4 files
14a4291 verified 7 months ago
  • omnipaint_insert.safetensors
    29.1 MB
    xet
    Upload 4 files 7 months ago
  • omnipaint_remove.safetensors
    29.1 MB
    xet
    Upload 4 files 7 months ago