 “BrianChen1129”
|
b0a104fef6
|
update
|
2025-07-05 13:32:16 -07:00 |
|
|
“BrianChen1129”
|
a01b6d386d
|
update
|
2025-07-05 13:32:16 -07:00 |
|
|
William Lin
|
bf4de1f38f
|
[chore] Upgrade min Python version from 3.8 to 3.10 (#597)
|
2025-07-04 22:08:31 -05:00 |
|
|
William Lin
|
66fdcc8e76
|
[Training] Use inference pipeline for training validation (#585)
|
2025-07-04 17:23:55 -05:00 |
|
|
Wenxuan Tan
|
ed1e8d6bad
|
[Feature] Offload all text encoders by default (#594)
|
2025-07-03 19:30:14 -05:00 |
|
|
Kevin Lin
|
b9423ca3f8
|
Add ComfyUI custom node for inference (#596)
|
2025-07-03 16:09:03 -05:00 |
|
|
Wenxuan Tan
|
ad16289871
|
[LoRA] Fix lora merge weights (#579)
|
2025-07-02 00:12:52 -05:00 |
|
|
Wenxuan Tan
|
2a41da1e6b
|
Fix VAE precisions (#588)
|
2025-07-01 14:49:04 -05:00 |
|
|
William Lin
|
32133171da
|
[chore] Release 0.1.1 (#592)
v0.1.1
|
2025-07-01 01:43:12 -05:00 |
|
|
Kevin Lin
|
19674c6f29
|
[CI] Fix fork builds (#590)
|
2025-07-01 01:03:34 -05:00 |
|
|
Yongqi Chen
|
508afb7002
|
[docs] Update Readme (#591)
|
2025-06-30 23:48:45 -05:00 |
|
|
Yongqi Chen
|
288ea88105
|
[Feat][Training] Rename weight conversion function and update gradient checkpoint in scripts (#589)
|
2025-07-01 00:20:02 -04:00 |
|
|
Jinzhe Pan
|
eb0f1318f3
|
[Feat] activation checkpointing (#584)
|
2025-06-30 15:24:29 -05:00 |
|
|
William Lin
|
ce9b5910cc
|
[Training] add caption to validation log (#582)
|
2025-06-30 02:42:17 -05:00 |
|
|
William Lin
|
d0e5a6214a
|
[misc] [training] Add --video_length_tolerance_range 10 to preprocessing scripts (#581)
|
2025-06-30 02:21:22 -05:00 |
|
|
Wenxuan Tan
|
834562b2db
|
[CI] Fix pre-commit CI (#578)
|
2025-06-29 16:52:29 -05:00 |
|
|
Wei (Will) Feng
|
060cc7b9ba
|
fully_shard usage on RMSNorm (#577)
|
2025-06-29 16:35:24 -05:00 |
|
|
Yongqi Chen
|
6c58a5ba62
|
[Bugfix]Fix VSA sp for training/inference (#574)
|
2025-06-29 13:44:33 -05:00 |
|
|
William Lin
|
48d9f61f86
|
[ci] [misc] fix training test threshold (#573)
|
2025-06-28 22:17:18 -05:00 |
|
|
William Lin
|
5f938b5844
|
[Revert] "[Feature] Load weights from distributed" (#571)
|
2025-06-28 20:55:14 -05:00 |
|
|
Wenxuan Tan
|
74da2a7370
|
Fix CLIP config (#568)
|
2025-06-28 19:01:23 -05:00 |
|
|
Kevin Lin
|
580d6dfe1f
|
[CI] Add tests to Modal (#562)
|
2025-06-28 14:02:16 -05:00 |
|
|
Wenxuan Tan
|
344e43006a
|
[CI] Fix SSIM and transformers CI (#564)
|
2025-06-28 00:26:20 -05:00 |
|
|
Wenxuan Tan
|
c5155b256e
|
[Feature] Load weights from distributed (#470)
|
2025-06-27 22:52:40 -05:00 |
|
|
William Lin
|
e005c7f3ac
|
[Docs] [Training] add readme for example training (#563)
|
2025-06-27 14:50:42 -05:00 |
|
|
Yongqi Chen
|
ff5a79ef60
|
[Feature][Inference] Add VSA inference script (#561)
|
2025-06-27 02:19:23 -05:00 |
|
|
William Lin
|
ab01dc4ba5
|
[Feature] [Training] Add i2v training (#559)
|
2025-06-27 01:56:50 -05:00 |
|
|
William Lin
|
285a950c1b
|
[CI] fix vae and ssim tests (#557)
|
2025-06-26 23:53:01 -05:00 |
|
|
William Lin
|
46a0a85d85
|
[Training] Fixes SP for training; Improve Datasets and schema (#555)
|
2025-06-26 21:13:28 -05:00 |
|
|
Yongqi Chen
|
4aeabbc629
|
[Feature][Training] Add cfg rate for dataset loader (#556)
|
2025-06-26 18:22:37 -04:00 |
|
|
Wenxuan Tan
|
949bb5c835
|
[CI] Fix CI checks (#553)
|
2025-06-25 14:07:51 -05:00 |
|
|
Wenxuan Tan
|
aab74c1271
|
[Kernel] Remove all syncs from STA & VSA kernels (#517)
|
2025-06-23 13:13:09 -07:00 |
|
|
Yongqi Chen
|
f89d86944f
|
[Feature][Training]Add diffusers format checkpoint saving for inference (#542)
|
2025-06-22 01:23:41 -04:00 |
|
|
William Lin
|
8741d204a5
|
[Training] Refactor and improve validation datasets (#539)
|
2025-06-21 17:58:35 -07:00 |
|
|
Wenxuan Tan
|
cdc85f58a8
|
[chore] Bump torch to 2.7.1 to support Blackwell (#483)
|
2025-06-20 22:10:56 -07:00 |
|
|
William Lin
|
0262d2f089
|
[misc] [training] Reorganize training pipeline (#533)
|
2025-06-20 20:42:25 -07:00 |
|
|
William Lin
|
62c0343465
|
[bugfix] [VSA] Fix layernorm type for VSA Wan2.1 TransformerBlock (#534)
|
2025-06-20 00:24:51 -07:00 |
|
|
William Lin
|
1e1a023fb0
|
[bugfix] Fix stage validator for multi text encoder models (#535)
|
2025-06-19 22:49:16 -07:00 |
|
|
William Lin
|
1d2517ad8e
|
[misc] Remove gradient checking code (#532)
|
2025-06-18 23:29:25 -07:00 |
|
|
William Lin
|
d41186cb4a
|
[Feat] Add Stage input and output verification (#523)
|
2025-06-18 23:29:11 -07:00 |
|
|
|
78e0c7eec9
|
Specify cu128 Pytorch installation (#530)
Co-authored-by: Edenzzzz <wtan45@wisc.edu>
Co-authored-by: Wenxuan Tan <wenxuan.tan@wisc.edu>
|
2025-06-18 20:02:50 -05:00 |
|
|
Wenxuan Tan
|
1c41a94b62
|
[Refactor] Move dict_to_3d_list under utils (#507)
|
2025-06-18 13:34:37 -07:00 |
|
|
Yongqi Chen
|
2e66aafe20
|
[Bugfix][Readme]Fix readme website bugs and add VSA finetune docs (#531)
|
2025-06-17 22:48:29 -07:00 |
|
|
Yongqi ChenandWill Lin
|
55074bda76
|
[CI] Add STA-inference/VSA-training test (#527)
Co-authored-by: Will Lin <wlsaidhi@gmail.com>
|
2025-06-17 21:13:06 -07:00 |
|
|
William Lin
|
de65bec2b7
|
[Ci] add sta and vsa install to docker image (#528)
|
2025-06-17 18:09:48 -07:00 |
|
|
Yongqi Chen
|
7664dd0de3
|
[Bugfix][Inference]Fix envs.attn_backend (#525)
|
2025-06-17 18:38:06 -05:00 |
|
|
William Linandkevin314
|
019a88ced4
|
[CI][bugfix] Use new 3.12 docker image (#526)
Co-authored-by: kevin314 <kevin.lin.cs1@gmail.com>
|
2025-06-17 15:37:08 -07:00 |
|
|
Kevin Lin
|
72de11abcc
|
[CI] Add current PR test workflow to Buildkite/Modal (#512)
|
2025-06-17 13:29:22 -07:00 |
|
|
Kevin Lin
|
d71a4ebffc
|
[CI] Update Docker image to flash-attn 2.8.0 / CUDA 12.8 (#524)
|
2025-06-16 17:48:23 -07:00 |
|
|
William Lin
|
1089ab43bf
|
[bugfix] [Training] use diffusers fp32layernorm for wan2.1 (#490)
|
2025-06-15 22:45:48 -07:00 |
|