William Lin 6300329cd5 [infra]: Add activation trace hooks for pipeline debugging (#1293)

2026-05-07 16:38:13 -07:00

..

__init__.py

[feat] Hooks API and layerwise offloading for all DiTs (#1006)

2026-01-17 11:22:02 -08:00

activation_trace.py

[infra]: Add activation trace hooks for pipeline debugging (#1293)

2026-05-07 16:38:13 -07:00

hooks.py

[feat] pre-commit support 120 col num (#1167)

2026-03-11 20:19:30 -07:00

layerwise_offload.py

[feat] pre-commit support 120 col num (#1167)

2026-03-11 20:19:30 -07:00