Kontext
Collection
8 items • Updated
import torch
from diffusers import DiffusionPipeline
from diffusers.utils import load_image
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("joyfox/Kontext-Remover-General-LoRA", dtype=torch.bfloat16, device_map="cuda")
prompt = "Remove this woman from the scene"
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]


This is a model for background removal, trained on black-forest-labs/FLUX.1-Kontext-dev, and is suitable for background removal tasks of e-commerce images, character images, and object images.For use in ComfyUI.
Remove xxx from the scene
Weights for this model are available in Safetensors format.
This model was trained by the AI Laboratory of Chongqing Valiant Cat Technology Co., LTD(https://vvicat.com/).Business cooperation is welcome
Base model
black-forest-labs/FLUX.1-Kontext-dev