Yuanheng Zhao 2f5e32477c [NFC] Remove redundant torch.no_grad in models and pipelines (#854) ...

Signed-off-by: yuanheng <jonathan.zhaoyh@gmail.com>

2026-01-27 12:36:54 +00:00

..

layers

[Chore] Bump Python version and apply formatting changes for v0.12 release (#126)

2025-12-16 15:19:40 +08:00

model_loader

[Chore] Bump Python version and apply formatting changes for v0.12 release (#126)

2025-12-16 15:19:40 +08:00

models

[NFC] Remove redundant torch.no_grad in models and pipelines (#854)

2026-01-27 12:36:54 +00:00

stage_configs

[Feature] Implement YuanrongConnector based on OmniConnectorBase (#716)

2026-01-27 20:17:02 +08:00

stage_input_processors

[Feature]Support async computation and communication across stages by chunks (#727)

2026-01-26 11:21:24 +08:00

__init__.py

update ar_dit_test

2025-09-28 21:58:48 +08:00

custom_process_mixin.py

[Core] customize process between stages, remove the need of "request_ids" in model forward, rebase to 0.12.0 (#458)

2025-12-24 13:26:33 +00:00