Kevin Lin bdfdf1dfee [Training] Add distributed checkpointing (#458)

2025-06-02 15:15:45 -07:00

..

attention

[bug] fix bs > 1 (#418)

2025-05-21 21:07:01 -07:00

configs

[Training] [5/n] Add single gpu training pipeline (#447)

2025-05-29 11:49:46 -07:00

dataset

Fix WanVideo (#461)

2025-06-01 23:08:11 -07:00

distributed

[Training] [5/n] Add single gpu training pipeline (#447)

2025-05-29 11:49:46 -07:00

entrypoints

[V1] Update where num_frame rounding is done (#403)

2025-05-12 11:52:49 -07:00

layers

[bug] fix bs > 1 (#418)

2025-05-21 21:07:01 -07:00

logging_utils

[do not merge] Rebased refactor (#270)

2025-03-29 17:43:47 -05:00

models

Fix WanVideo (#461)

2025-06-01 23:08:11 -07:00

pipelines

Fix WanVideo (#461)

2025-06-01 23:08:11 -07:00

platforms

[V1] Remove vLLM dependency (#413)

2025-05-18 00:57:28 -07:00

tests

[V1] Remove vLLM dependency (#413)

2025-05-18 00:57:28 -07:00

third_party

[do not merge] Rebased refactor (#270)

2025-03-29 17:43:47 -05:00

training

[Training] Add distributed checkpointing (#458)

2025-06-02 15:15:45 -07:00

worker

Fulfill worker response on interrupt (#417)

2025-05-21 20:59:11 -07:00

__init__.py

[Model] Remove RMSNorm's forward_native hardcode from Wan (#339)

2025-04-14 16:46:43 -07:00

envs.py

[Attn] Add SageAttention Backend (#366)

2025-04-28 12:40:07 -06:00

fastvideo_args.py

[Training] Add distributed checkpointing (#458)

2025-06-02 15:15:45 -07:00

forward_context.py

Add Teacache to V1 (#371)

2025-05-06 11:58:56 -07:00

logger.py

[V1] Gradio demo with new API (#357)

2025-04-19 18:14:24 -07:00

utils.py

misc: add remote pdb for debugging workers (#456)

2025-05-30 20:44:27 -05:00