valiantcat Qwen-Image-Edit LoRA

Model Card for Model ID

This is a model for object removal, trained on ``Qwen/Qwen-Image-Edit`, and is suitable for object removal tasks of e-commerce images, character images, and object images.For use in `ComfyUI``.

The greatest advantage of using this LORA is that it maintains the consistency of the original image without changing any parts.

ComfyUI Workflow

This LoRA works with a modified version of Comfy's Qwen-Image-Edit workflow. The main modification is adding a Qwen-Image-Edit LoRA node connected to the base model.

See the Downloads section above for the modified workflow.

Direct Use

from diffusers import QwenImageEditPipeline
import torch
from PIL import Image

# Load the pipeline
pipeline = QwenImageEditPipeline.from_pretrained("Qwen/Qwen-Image-Edit")
pipeline.to(torch.bfloat16)
pipeline.to("cuda")

# Load trained LoRA weights for in-scene editing
pipeline.load_lora_weights("valiantcat/Qwen-Image-Edit-Remover-General-LoRA",weight_name="qwen-edit-remover.safetensors")

# Load input image
image = Image.open("./result/test.png").convert("RGB")

# Define in-scene editing prompt
prompt = "移除猫猫"

# Generate edited image with enhanced scene understanding
inputs = {
    "image": image,
    "prompt": prompt,
    "generator": torch.manual_seed(12345),
    "true_cfg_scale": 4.0,
    "negative_prompt": " ",
    "num_inference_steps": 50,
}

with torch.inference_mode():
    output = pipeline(**inputs)
    output_image = output.images[0]
    output_image.save("restlt.png")

Trigger phrase

There is no fixed trigger word. The specific removal prompt needs to be tested more

## Download model

Weights for this model are available in Safetensors format.

[Download](https://huggingface.co/valiantcat/Qwen-Image-Edit-Remover-General-LoRA)

## Training at Chongqing Valiant Cat

This model was trained by the AI Laboratory of Chongqing Valiant Cat Technology Co., LTD(```https://vvicat.com/```).Business cooperation is welcome