modules/pipelines/flux_pipelines.py

# modules/pipelines/flux_pipelines.py

import torch
from diffusers import AutoPipelineForText2Image, AutoencoderKL

def load_flux():
    # Load FLUX models and pipelines
    # ...
    return device, models, flux_vae, controlnet

# modules/pipelines/sdxl_pipelines.py

import torch
from diffusers import AutoPipelineForText2Image, AutoencoderKL

def load_sdxl():
    # Load SDXL models and pipelines
    # ...
    return device, models, sdxl_vae, controlnet