|
|
--- |
|
|
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: >- |
|
|
remove all the clothes of the figure in the picture |
|
|
output: |
|
|
url: result/result1.png |
|
|
- text: >- |
|
|
remove all the clothes of the figure in the picture |
|
|
output: |
|
|
url: result/result2.png |
|
|
- text: >- |
|
|
remove all the clothes of the figure in the picture |
|
|
output: |
|
|
url: result/result3.png |
|
|
--- |
|
|
# starsfriday 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 it is mainly used to remove clothes from characters.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/Jonny001/Qwen-Image-Edit-Remove-Clothes/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("Jonny001/Qwen-Image-Edit-Remove-Clothes",weight_name="qwen-edit-remove-clothes.safetensors") |
|
|
|
|
|
# Load input image |
|
|
image = Image.open("./result/test.jpg").convert("RGB") |
|
|
|
|
|
# Define in-scene editing prompt |
|
|
prompt = "remove all the clothes of the figure in the picture " |
|
|
|
|
|
# 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 |
|
|
|
|
|
```remove all the clothes of the figure in the picture``` |
|
|
|
|
|
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/Jonny001/Qwen-Image-Edit-Remove-Clothes) |
|
|
|
|
|
## 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 |
