thibaud

/

sdxl_dpo_turbo

like 83

    print("Loading DPO turbo model...")
    pipe = StableDiffusionXLPipeline.from_single_file(
        "models/sdxl_dpo_turbo.safetensors", 
        torch_dtype=torch.float16, 
        safety_checker=None
    ) 
    pipe.safety_checker = None
    print("DPO turbo model loaded.")
    # pipe.scheduler = DPMSolverSDEScheduler.from_config(pipe.scheduler.config, use_karras_sigmas='true')
    pipe.enable_model_cpu_offload()

    return pipe