"""Strict native loader for the Wan2.2 VAE shipped by Lance."""
from pathlib import Path
import torch
from native.vae2_2 import Wan2_2_VAE

def load_model(root, *, device='cpu', dtype=torch.float32):
    root = Path(root)
    return Wan2_2_VAE(vae_pth=str(root / 'model.pth'), device=device, dtype=dtype)