script.py

from pipeline import load_pipeline_assets, sample_batch


def main() -> None:
    """Example script showing how to invoke the diffusion pipeline."""
    load_pipeline_assets()
    print("✅ Model loaded successfully!")

    denoiser_state, history = sample_batch(
        "ddim",
        n_steps=10,
        n_samples=5,
        seed=456,
        keep_history=True,
    )

    print(f"state.position shape: {denoiser_state.integrator_state.position.shape}")
    print(f"History length: {len(history)}")


if __name__ == "__main__":
    main()