import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Wan-AI/Wan2.1-I2V-14B-720P", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("Estebandidoooi/NSFW")
prompt = "porn, sex, pose, pussy, ass, boobs."
image = pipe(prompt).images[0]Not-For-All-Audiences
This repository has been marked as containing sensitive content and may contain potentially harmful and sensitive information.
View model card