import torch
from diffusers import AutoPipelineForText2Image

def load_pipeline():
    pipe = AutoPipelineForText2Image.from_pretrained(
        "./rimuruXL",
        torch_dtype=torch.float16
    )
    pipe.to("cuda" if torch.cuda.is_available() else "cpu")
    pipe.safety_checker = None  # Disable NSFW filter (opsional)
    return pipe