from diffusers import DiffusionPipeline
import torch

model_name = "Qwen/Qwen-Image"

# Load the pipeline
if torch.cuda.is_available():
    torch_dtype = torch.bfloat16
    device = "cuda"
else:
    torch_dtype = torch.float32
    device = "cpu"

pipe = DiffusionPipeline.from_pretrained(model_name, torch_dtype=torch_dtype)
pipe = pipe.to(device)

# Load LoRA weights
pipe.load_lora_weights('starsfriday/Qwen-Image-NSFW/qwen_image_nsfw.safetensors', adapter_name="lora")

prompt = '''rsq, The image depicts a young woman sitting on the ground with her legs spread apart, in what appears to be an outdoor setting. She has long brown hair and is smiling at the camera. The background shows greenery, suggesting she might be in a park or garden. Her pose and expression convey confidence and comfort.   '''
negative_prompt =  " "
image = pipe(
    prompt=prompt,
    negative_prompt=negative_prompt,
    width=1024,
    height=1024,
    num_inference_steps=50,
    true_cfg_scale=5,
    generator=torch.Generator(device="cuda").manual_seed(123456)
)
image = image.images[0]

image.save("output.png")