| 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 |