How to use from the
Use from the
Diffusers library
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline

# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("Heouzen/LoKr_WAN22_T2V_A14B_sks_thea_r64")

prompt = "sks_thea, The image shows a person with hijab, wearing a maroon sweater. The person is resting their head on their hand, with their eyes looking directly at the camera. The background includes a white wall with a small air conditioning unit mounted above it. There is also a wooden door visible to the left, and a small plant with red flowers on a shelf or table in the upper right corner. The lighting suggests it is daytime, with natural light illuminating the scene. --num_frames 1"
image = pipe(prompt).images[0]

LoKr_WAN22_T2V_A14B_sks_thea_r64

Model trained with AI Toolkit by Ostris

Prompt
sks_thea, The image shows a person with hijab, wearing a maroon sweater. The person is resting their head on their hand, with their eyes looking directly at the camera. The background includes a white wall with a small air conditioning unit mounted above it. There is also a wooden door visible to the left, and a small plant with red flowers on a shelf or table in the upper right corner. The lighting suggests it is daytime, with natural light illuminating the scene. --num_frames 1
Prompt
A medium-shot portrait captures a smiling young woman, sks_thea, posing with a cheerful expression while giving a thumbs-up gesture. She has a fair complexion, dark eyebrows, and visible teeth while smiling. Her head is covered in a dark, wrapped hijab style with a prominent cross-over design at the forehead. She is wearing a dark charcoal gray denim jacket layered over a black base garment. The background shows a simple room interior with a light-colored wall, a circular analog wall clock, and a hint of a mirror and closet door, which keeps the focus entirely on the subject. Her relaxed body language and bright, friendly demeanor are central to her appearance, with the textured material of the denim jacket adding detail to her ensemble.
Prompt
A medium shot captures sks_thea posing with a playful, aggressive expression, her brows furrowed and lips pursed while she holds both fists raised in front of her chest as if ready to fight. She is wearing a black hijab styled with a wrapped turban-like headband that frames her face. Her outfit consists of a dark-wash denim jacket layered over a black base, creating a casual yet bold aesthetic. The lighting is soft and even, highlighting her expressive features and the texture of the denim. The background is partially blurred, showing elements of an indoor room with a wall clock and a mirror, which do not distract from her prominent pose. Her focused gaze and defiant stance are the central elements of this portrait. --num_frames 1

Trigger words

You should use sks_thea to trigger the image generation.

Download model and use it with ComfyUI, AUTOMATIC1111, SD.Next, Invoke AI, etc.

Weights for this model are available in Safetensors format.

Download them in the Files & versions tab.

Use it with the 🧨 diffusers library

from diffusers import AutoPipelineForText2Image
import torch

pipeline = AutoPipelineForText2Image.from_pretrained('ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16', torch_dtype=torch.float16).to('cuda')
pipeline.load_lora_weights('Heouzen/LoKr_WAN22_T2V_A14B_sks_thea_r64', weight_name='LoKr_WAN22_T2V_A14B_sks_thea_r64.safetensors')
image = pipeline('sks_thea, The image shows a person with hijab, wearing a maroon sweater. The person is resting their head on their hand, with their eyes looking directly at the camera. The background includes a white wall with a small air conditioning unit mounted above it. There is also a wooden door visible to the left, and a small plant with red flowers on a shelf or table in the upper right corner. The lighting suggests it is daytime, with natural light illuminating the scene. --num_frames 1').images[0]
image.save("my_image.png")

For more details, including weighting, merging and fusing LoRAs, check the documentation on loading LoRAs in diffusers

Downloads last month
36
Inference Examples
Examples
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for Heouzen/LoKr_WAN22_T2V_A14B_sks_thea_r64

Adapter
(54)
this model