custom_pipeline.py

# Custom FLUX pipeline with CFG support
# This is a simplified version - for full implementation, see the original repo

from diffusers import FluxPipeline

class FluxWithCFGPipeline(FluxPipeline):
    """FLUX pipeline with CFG support for better control"""
    pass  # Use base FluxPipeline for now - can be extended later