updated readme. c44b6b1
Bryce commited on
How to use Notid/qwen-edit-2511-mask-extraction with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2511", torch_dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("Notid/qwen-edit-2511-mask-extraction")
prompt = "Create a black and white alpha mask for the object with a red outline"
image = pipe(prompt).images[0]