import torch
from diffusers import QwenImageImg2ImgPipeline
from diffusers.utils import load_image
pipe = QwenImageImg2ImgPipeline.from_pretrained("Qwen/Qwen-Image", torch_dtype=torch.bfloat16,)
pipe = pipe.to("cuda")
init_image = load_image("https://github.com/Trgtuan10/Image_storage/blob/main/cute_cat.png?raw=true")
prompt = "wizard dog, Gandalf-inspired, Lord of the Rings aesthetic, majestic yet cute, Studio Ghibli style"
negative_prompt = ""
strengths = [0.6, 0.7, 0.8, 0.9, 1.0]
for s in strengths:
 out = pipe(
 prompt=prompt,
 negative_prompt=negative_prompt,
 image=init_image,
 height=init_image.size[1],
 width=init_image.size[0],
 strength=s,
 num_inference_steps=35,
 true_cfg_scale=4.0,
 generator=torch.Generator(device="cuda").manual_seed(42),
 )
 out.images[0].save(f"example_img2img_{s}.png")

import torch
from PIL import Image
from diffusers import QwenImageInpaintPipeline
from diffusers.utils import load_image
pipe = QwenImageInpaintPipeline.from_pretrained("Qwen/Qwen-Image", torch_dtype=torch.bfloat16)
pipe.to("cuda")
prompt = "cat wizard with red hat, gandalf, lord of the rings, detailed, fantasy, cute, adorable, Pixar, Disney"
negative_prompt = " "
source = load_image("https://github.com/Trgtuan10/Image_storage/blob/main/cute_cat.png?raw=true")
mask = load_image("https://github.com/Trgtuan10/Image_storage/blob/main/mask_cat.png?raw=true")
strengths = [0.6, 0.7, 0.8, 0.9, 1.0]
for strength in strengths:
 image = pipe(
 prompt=prompt,
 negative_prompt=negative_prompt,
 height=source.size[1],
 width=source.size[0],
 image=source,
 mask_image=mask,
 strength=strength,
 num_inference_steps=35,
 true_cfg_scale=4.0,
 generator=torch.Generator(device="cuda").manual_seed(42)
 ).images[0]
 image.save(f"qwen_inpainting_{strength}.png")