Skip to content

API Reference

Breaking Changes in v0.2

Recent upgrade to version >= v0.2.0 may cause some breaking changes. Make some documented instructions may be outdated. You can install a specific version by pip install fusion-bench==0.1.6 or checkout to a specific version by git checkout v0.1.6. If you encounter any issues, please feel free to raise an issue. We are working on the documentation and will update it as soon as possible.

Use version >=0.2.0 is recommended.

Here we provides an overview of the API reference for FusionBench.

Entry Points

main(cfg)

Source code in fusion_bench/scripts/cli.py 
31
32
33
34
35
36
37
38
39
@hydra.main(
    config_path=_get_default_config_path(),
    config_name="fabric_model_fusion",
    version_base=None,
)
def main(cfg: DictConfig) -> None:
    OmegaConf.resolve(cfg)
    program: BaseHydraProgram = instantiate(cfg)
    program.run()

Class Definitions

Base class for all FusionBench components.

Modules