import torch
import spaces
from diffusers import DiffusionPipeline, AutoencoderTiny, FluxImg2ImgPipeline
from huggingface_hub import hf_hub_download


def feifeimodload():

    dtype = torch.bfloat16
    device = "cuda" if torch.cuda.is_available() else "cpu"

    pipe = DiffusionPipeline.from_pretrained(
        "aifeifei798/DarkIdol-flux-v1.2", torch_dtype=dtype
    ).to(device)
    
    pipe.vae.enable_slicing()
    pipe.vae.enable_tiling()
    torch.cuda.empty_cache()
    return pipe