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