JerryZhou54 42a2f272d5 Preprocessing Stage: 1. Save to parquets periodically 2. Allow resuming from the middle 3. Doesn't support multi-gpu for now. Data Loader Stage: 1. Allow multi-gpu dataloader 2. Doesn't drop files even if number of parquet files is not divisible by num_gpus 3. Able to resume training

2025-05-23 00:57:39 +00:00

..

data_preprocess

Preprocessing Stage: 1. Save to parquets periodically 2. Allow resuming from the middle 3. Doesn't support multi-gpu for now. Data Loader Stage: 1. Allow multi-gpu dataloader 2. Doesn't drop files even if number of parquet files is not divisible by num_gpus 3. Able to resume training

2025-05-23 00:57:39 +00:00

distill

[Feat] Add STA for StepVideo (#200)

2025-02-20 11:33:58 -08:00

models

[Misc] Replace instances of time.time() with time.perf_counter() (#396)

2025-05-11 15:56:39 -07:00

sample

Added multi-GPU support for Hunyuan STA (#211)

2025-02-21 14:16:28 -08:00

utils

[FIX] fix isort format (#203)

2025-02-20 12:20:20 -08:00

v1

Preprocessing Stage: 1. Save to parquets periodically 2. Allow resuming from the middle 3. Doesn't support multi-gpu for now. Data Loader Stage: 1. Allow multi-gpu dataloader 2. Doesn't drop files even if number of parquet files is not divisible by num_gpus 3. Able to resume training

2025-05-23 00:57:39 +00:00

__init__.py

[V1] Model config (#358)

2025-04-30 14:48:22 -07:00

distill_adv.py

[Misc] Replace instances of time.time() with time.perf_counter() (#396)

2025-05-11 15:56:39 -07:00

distill_wan.py

Add data preprocessing script for WAN

2025-05-19 15:09:26 +00:00

distill.py

[Misc] Replace instances of time.time() with time.perf_counter() (#396)

2025-05-11 15:56:39 -07:00

train.py

[Misc] Replace instances of time.time() with time.perf_counter() (#396)

2025-05-11 15:56:39 -07:00