Nsfw
Collection
2 items • Updated
import torch
from diffusers import DiffusionPipeline
from diffusers.utils import load_image, export_to_video
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("undefined,lkzd7/WAN2.2_LoraSet_NSFW", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("KLOWNZone23/lora")
prompt = "TOT"
input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png")
image = pipe(image=input_image, prompt=prompt).frames[0]
export_to_video(output, "output.mp4")My lora model
You should use TOT to trigger the image generation.
Weights for this model are available in Safetensors format.
Download them in the Files & versions tab.
Training was done using fal.ai/models/fal-ai/turbo-flux-trainer.
Base model
lkzd7/WAN2.2_LoraSet_NSFW