 Will Lin
|
ec6ba01eb1
|
move utils into training_utils
|
2025-05-28 11:40:15 -07:00 |
|
|
Zihang-He
|
086c9600a8
|
added gradient clipping
|
2025-05-28 11:38:55 -07:00 |
|
|
Will Lin
|
5630eb5041
|
gradient checking
|
2025-05-28 11:37:46 -07:00 |
|
|
Will Lin
|
31f62ab249
|
cleanup
|
2025-05-28 11:36:36 -07:00 |
|
|
Will Lin
|
9db9c84cc9
|
add validation
|
2025-05-28 11:36:34 -07:00 |
|
|
JerryZhou54
|
ab9eac9ee8
|
Small fix
|
2025-05-28 11:35:34 -07:00 |
|
|
JerryZhou54
|
3cce02193d
|
Small fix
|
2025-05-28 11:35:32 -07:00 |
|
|
JerryZhou54
|
a1a42f7afc
|
Integrate the new parquet dataloader into training pipeline
|
2025-05-28 11:35:07 -07:00 |
|
|
|
d36c209191
|
Will/training (#425)
Co-authored-by: Kevin Lin <42618777+kevin314@users.noreply.github.com>
Co-authored-by: William Lin <SolitaryThinker@users.noreply.github.com>
Co-authored-by: Will Lin <wlsaidhi@gmail.com>
|
2025-05-28 11:34:17 -07:00 |
|
|
JerryZhou54
|
7f4379acc0
|
Finish data preprocessing and loading
|
2025-05-28 11:31:41 -07:00 |
|
|
JerryZhou54
|
7ba8cdf148
|
Add data preprocessing script for WAN
|
2025-05-28 11:27:39 -07:00 |
|
|
|
7ab32539af
|
[Training] [1/n] Add latent datasets (#438)
Co-authored-by: Wei Zhou <wzhou322@gatech.edu>
Co-authored-by: JerryZhou54 <zhouw.jerry2017@outlook.com>
Co-authored-by: “BrianChen1129” <yongqich@umich.edu>
|
2025-05-28 11:01:52 -07:00 |
|
|
William Lin
|
6ef8fcb61d
|
[Training] [4/n] add training save checkpoint (#441)
|
2025-05-27 17:53:53 -07:00 |
|
|
William Lin
|
016e24da63
|
[Training] [3/n] Add training args and dependencies (#440)
|
2025-05-27 17:53:39 -07:00 |
|
|
William Lin
|
85b8717545
|
[Training] [2/n] add bwd for all2all and all_gather (#439)
|
2025-05-27 14:27:54 -07:00 |
|
|
Wenxuan Tan
|
657fd745e1
|
misc: Trigger transformers CI for layers and attention code change (#434)
|
2025-05-27 11:43:23 -07:00 |
|
|
applesaucethebunandBrayden Zhong
|
12647457a7
|
[Misc] Small fixes to Torch code (#395)
Co-authored-by: Brayden Zhong <b8zhong@uwaterloo.ca>
|
2025-05-23 14:40:24 -07:00 |
|
|
Kevin Lin
|
298f74f956
|
Set device for encode (#420)
|
2025-05-23 14:19:45 -07:00 |
|
|
Wenxuan Tan
|
ee8babb298
|
Unify env report script in issue template (#423)
|
2025-05-23 14:19:18 -07:00 |
|
|
Wenxuan Tan
|
60295cc03f
|
Use version.py (#424)
|
2025-05-23 14:17:33 -07:00 |
|
|
William Lin
|
1572e13b6e
|
[Tests] don't run 3.10 and 3.11 for SSIM (#427)
|
2025-05-23 12:59:50 -07:00 |
|
|
Wenxuan Tan
|
a157275b4c
|
Fix version number (#422)
|
2025-05-22 12:37:34 -07:00 |
|
|
William Lin
|
c4dbe7dac3
|
[bug] fix bs > 1 (#418)
|
2025-05-21 21:07:01 -07:00 |
|
|
Kevin Lin
|
d39591108e
|
Fulfill worker response on interrupt (#417)
|
2025-05-21 20:59:11 -07:00 |
|
|
William Lin
|
ace6e971e5
|
[V1] Remove vLLM dependency (#413)
|
2025-05-18 00:57:28 -07:00 |
|
|
William Lin
|
b4f6758253
|
[Teacache] allow None for forward_context batch when using teacache (#412)
|
2025-05-17 18:43:14 -07:00 |
|
|
William Lin
|
535d29b392
|
[Docs] Fix image (#407)
|
2025-05-12 14:30:05 -07:00 |
|
|
William Lin
|
b4255517e0
|
[Docs] Add CLI docs (#406)
|
2025-05-12 14:16:48 -07:00 |
|
|
William Lin
|
6eeb60613f
|
Release 0.1.0 (#405)
v0.1.0
|
2025-05-12 11:54:23 -07:00 |
|
|
William Lin
|
53d2c7791f
|
[V1] Update where num_frame rounding is done (#403)
|
2025-05-12 11:52:49 -07:00 |
|
|
William Lin
|
53cb693dca
|
[V1] Docs Update (#402)
|
2025-05-12 11:52:09 -07:00 |
|
|
Kevin Lin
|
6f72d24876
|
[CLI] Default to pipeline config (#401)
|
2025-05-12 00:22:53 -07:00 |
|
|
William Lin
|
d1459e9976
|
[V1] Update README (#400)
|
2025-05-11 22:32:03 -07:00 |
|
|
William Lin
|
59ab481eb1
|
release 0.0.5 (#399)
v0.0.5
|
2025-05-11 16:23:11 -07:00 |
|
|
William Lin
|
0cf001986a
|
[Docs] More docs update (#394)
|
2025-05-11 16:19:47 -07:00 |
|
|
applesaucethebunandBrayden Zhong
|
51956369a5
|
[Misc] Replace instances of time.time() with time.perf_counter() (#396)
Co-authored-by: Brayden Zhong <b8zhong@uwaterloo.ca>
|
2025-05-11 15:56:39 -07:00 |
|
|
Kevin Lin
|
4b0970cbbf
|
[CI] Set volume_size required to false (#398)
|
2025-05-11 15:55:34 -07:00 |
|
|
Kevin Lin
|
94bf47a572
|
[CI] Set default disk size (#397)
|
2025-05-11 15:34:17 -07:00 |
|
|
River (Zihang He)andWill Lin
|
1a3ac9074b
|
Zihang stepvideo v1 (#389)
Co-authored-by: Will Lin <wlsaidhi@gmail.com>
|
2025-05-11 14:41:42 -07:00 |
|
|
Kevin Lin
|
fb0581d5b0
|
[CLI] Update cli to support new api/model config (#384)
|
2025-05-11 13:50:33 -07:00 |
|
|
Kevin Lin
|
6c74ab4132
|
[CI] Use python 3.10/3.11 for SSIM test (#392)
|
2025-05-10 13:48:29 -07:00 |
|
|
William Lin
|
9a91021c56
|
[Docs] Add collect_env.py and various docs update (#393)
|
2025-05-10 13:42:42 -07:00 |
|
|
Wei Zhou
|
51c94d6a73
|
[Misc] Small Fixes & Features (#390)
|
2025-05-07 15:40:49 -07:00 |
|
|
William Lin
|
dba38dbc03
|
Release 0.0.4 (#388)
|
2025-05-07 02:40:42 -07:00 |
|
|
William Lin
|
2034cc3c4f
|
[misc] Improve worker cleanup (#387)
|
2025-05-06 22:45:22 -07:00 |
|
|
William Lin
|
c69afce2f6
|
[Docs] Update for V1 (#381)
|
2025-05-06 16:34:44 -07:00 |
|
|
Wei Zhou
|
b08e758eb3
|
Small Fixes & Features (#378)
|
2025-05-06 15:52:34 -07:00 |
|
|
William Lin
|
3f3462d7ce
|
Cleanup Teacache params (#386)
|
2025-05-06 15:52:05 -07:00 |
|
|
William Lin
|
c9c47dd89c
|
Add Teacache to V1 (#371)
|
2025-05-06 11:58:56 -07:00 |
|
|
William Lin
|
9c4ef7c2f1
|
[Lint] fix (#382)
|
2025-05-06 00:42:55 -07:00 |
|