| --- |
| license: apache-2.0 |
| language: |
| - en |
| base_model: |
| - Qwen/Qwen-Image-Edit |
| tags: |
| - image-generation |
| - lora |
| - Qwen-Image |
| pipeline_tag: image-to-image |
| library_name: diffusers |
| widget: |
| - text: >- |
| 移除猫猫 |
| output: |
| url: result/result1.png |
| - text: >- |
| 同时移除这个小男孩和自行车 |
| output: |
| url: result/result2.png |
| - text: >- |
| 完全移除这个女人 |
| output: |
| url: result/result3.png |
| --- |
| # valiantcat Qwen-Image-Edit LoRA |
|
|
| <Gallery /> |
|
|
| ## Model Card for Model ID |
|
|
| <!-- Provide a quick summary of what the model is/does. --> |
|
|
| 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. |
|
|
| <div style="background-color: white; padding: 15px; border-radius: 8px; margin: 15px 0; box-shadow: 0 2px 4px rgba(0,0,0,0.1);"> |
| <h2 style="color: #24292e; margin-top: 0;">ComfyUI Workflow</h2> |
| <p>This LoRA works with a modified version of <a href="https://huggingface.co/valiantcat/Qwen-Image-Edit-Remover-General-LoRA/blob/main/Qwen-Edit-LORA.json" style="color: #0366d6; text-decoration: none;">Comfy's Qwen-Image-Edit workflow</a>. The main modification is adding a Qwen-Image-Edit LoRA node connected to the base model.</p> |
| <p>See the Downloads section above for the modified workflow.</p> |
| </div> |
|
|
| ### 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 |
|
|
| ```从场景中移除XXX``` |
|
|
| 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 |