src/diffsynth_fov/__init__.py

"""Foveation-specific additions on top of upstream DiffSynth-Studio.

These files are ported from the author's DiffSynth-Studio fork:
  - pipeline.py : Flux2FoveatedImagePipeline (mixed-resolution FLUX2 inference / training forward)
  - dit.py      : Flux2DiTFoveated (CRPA attention DiT)
  - loss.py     : FoveatedFlowMatchSFTLoss + helpers
  - logger.py   : WandbModelLogger with foveated validation visualization
  - runner.py   : training launcher with --max_training_steps support
"""

from .dit import Flux2DiTFoveated
from .logger import WandbModelLogger
from .loss import FoveatedFlowMatchSFTLoss
from .pipeline import Flux2FoveatedImagePipeline
from .runner import launch_data_process_task, launch_training_task