transformer_lens.tools.model_registry.generate_report module

Generate a markdown report of supported and unsupported models.

This script generates a comprehensive report showing: - All supported model IDs grouped by architecture - Total count of supported models - Unsupported architectures with model counts and descriptions

Usage:

python -m transformer_lens.tools.model_registry.generate_report python -m transformer_lens.tools.model_registry.generate_report –output custom_report.md python -m transformer_lens.tools.model_registry.generate_report –help

transformer_lens.tools.model_registry.generate_report.generate_report(output_path: Path | None = None) → str

Generate the markdown report.

Parameters:

output_path – Optional path to write the report. If None, only returns the string.

Returns:

The generated markdown report as a string.

transformer_lens.tools.model_registry.generate_report.get_architecture_description(arch_id: str) → str

Get a description for an architecture, with fallback.

transformer_lens.tools.model_registry.generate_report.main()

CLI entry point.