RandNMR73
eddc7efc64
inference works after changes added
...
new branch
Stop backprop through kv_cache
Enable timestep warping & using SelfForcing scheduler
checkpoint
text preprocessing ready
fix rotary embedding
Change Wan DiT to have 0 numerical diff with SF's Wan
2025-09-22 02:29:16 +00:00
31f000d1cc
[self-forcing] [5/n] Add Self-Forcing distillation pipeline ( #808 )
...
Co-authored-by: RandNMR73 <notomatthew31@gmail.com >
Co-authored-by: SolitaryThinker <wlsaidhi@gmail.com >
2025-09-20 19:32:10 -07:00
William Lin
c5f9ea53b2
[self-forcing] [4/n] Preprocessing for collecting ODE trajectory ( #788 )
2025-09-15 17:54:42 -07:00
William Lin
d32a7184da
[bugfix] Wan2.2 Boundary ratio ( #804 )
2025-09-15 11:17:35 -07:00
Wenxuan Tan and gemini-code-assist[bot]
2930abe456
[Bugfix] Fix VMoba requirements ( #802 )
...
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
2025-09-14 18:28:52 -07:00
William Lin
b93ef4289d
[bugfix] Fix empty PipelineConfigs for Wan2.2 A14B ( #800 )
2025-09-13 17:31:38 -07:00
401bdbd316
[self-forcing] [3/n] Text embed only preprocessing ( #797 )
...
Co-authored-by: RandNMR73 <notomatthew31@gmail.com >
Co-authored-by: JerryZhou54 <zhouw.jerry2017@outlook.com >
Co-authored-by: kevin314 <kevin.lin.cs1@gmail.com >
2025-09-13 14:03:53 -07:00
William Lin
1048d79cf8
[bugfix] pin gradio version and set current_vsa_sparsity in TrainingPipeline ( #798 )
2025-09-11 17:04:47 -07:00
1e8406162d
[bugfix] Fix delta calculation ( #796 )
...
Co-authored-by: zbchu2 <zbchu2@iflytek.com >
Co-authored-by: William Lin <SolitaryThinker@users.noreply.github.com >
2025-09-11 16:31:23 -07:00
William Lin
03edd35c83
[preprocessing] [self-forcing] [2/n] Improve preprocessing and add ode trajectory dataset schema ( #794 )
2025-09-10 17:33:57 -07:00
William Lin and RandNMR73
ac11127397
[Self-forcing] [1/n] Handle extra dim in time embedding and add timestep warping ( #792 )
...
Co-authored-by: RandNMR73 <notomatthew31@gmail.com >
2025-09-09 02:52:02 -07:00
Eric Liang and EricLiang
e028dcc7c0
[Backend][Vmoba] Add implementation of VMoba ( #778 )
...
Co-authored-by: EricLiang <https://github.com/EricLina >
2025-09-08 23:53:25 -07:00
Wenxuan Tan and gemini-code-assist[bot]
076f45c1ee
[Feature] Support Lora for DMD ( #755 )
...
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
2025-09-08 14:18:21 -07:00
85eb7265db
fix: lora_B init zeros ( #781 )
...
Co-authored-by: zbchu2 <zbchu2@iflytek.com >
Co-authored-by: Wenxuan Tan <wenxuan.tan@wisc.edu >
2025-09-05 22:56:52 -07:00
William Lin
d1e7aa0abd
[CI] Add ssim test for causal inference ( #784 )
2025-09-05 01:23:01 -07:00
William Lin
2d846c55a1
[misc] Improve text encoding stage ( #774 )
2025-09-04 17:51:27 -07:00
Jinzhe Pan
b318063c0a
[Preprocess][Fix] video quality issue ( #773 )
2025-09-03 20:47:33 -07:00
Jinzhe Pan
4aa307be55
[Preprocess][Feat] support torchvision to load video in new preprocessing ( #761 )
2025-09-01 23:37:01 -07:00
William Lin
2dcc5ea4f6
[chore] Release 0.1.6 ( #768 )
2025-08-28 20:56:21 -07:00
Wei Zhou and SolitaryThinker
359151d9a0
[Feature] Add wan2.2 5b i2v ( #760 )
...
Co-authored-by: SolitaryThinker <wlsaidhi@gmail.com >
2025-08-28 18:15:59 -07:00
Wei Zhou and SolitaryThinker
ce67cd3729
[Feat] Support Self-Forcing's Causal Inference for Wan2.1 T2V 1.3B ( #766 )
...
Co-authored-by: SolitaryThinker <wlsaidhi@gmail.com >
2025-08-28 16:47:49 -07:00
William Lin
663ea33ff1
[bugfix] Fix wrong HF model string for FastWan2.2 5B ( #763 )
2025-08-26 22:05:40 -07:00
William Lin
3ef04f1654
[misc] [docs] Various fixes for logging and docs ( #758 )
2025-08-23 21:13:50 -07:00
Jinzhe Pan
0eced76a41
[Feat][Preprocess] support multi-gpus ( #753 )
2025-08-23 11:34:42 +08:00
Jinzhe Pan
3ab6470d1a
[Feat][Preprocess] support merged dataset ( #752 )
2025-08-22 15:29:33 -07:00
Wenxuan Tan
989a03532c
Optionally use unmerged weights for inference ( #745 )
2025-08-22 15:20:31 -07:00
William Lin
fa15369a02
[bugfix] Check that model_index.json module is in required_modules list before removing ( #756 )
2025-08-22 14:36:44 -07:00
Zhang Peiyuan
78a9cb88d8
[Fix] fix seed in dmd denoising loop ( #736 )
2025-08-21 18:06:16 -07:00
Peng Xiao and 肖鹏
a0bff12746
[bugfix] [dmd] Align backward simulation with dmd2 sample back ( #744 )
...
Co-authored-by: 肖鹏 <xiaopeng1@aishi.ai >
2025-08-20 22:25:33 -07:00
Jinzhe Pan
911a6a6a35
[Feat][Preprocessing] i2v preprocessing workflow ( #737 )
2025-08-14 20:47:25 -07:00
Jinzhe Pan
7e7a0dba9d
feat: preprocess validation dataset only when exist ( #734 )
2025-08-12 02:16:31 -07:00
William Lin
6ceb4942a0
[bugfix] [dmd] Fix backward simulation and also naming in wan_i2v_dmd_pipeline ( #731 )
2025-08-10 21:13:30 -07:00
William Lin and RandNMR73
8cae5e4708
[feature] add Gradio live serving demo code ( #727 )
...
Co-authored-by: RandNMR73 <notomatthew31@gmail.com >
2025-08-10 15:34:03 -07:00
William Lin
2a773fa34e
[bugfix] [distill] remove i2v validation schema import in distill ( #728 )
2025-08-09 20:47:42 -07:00
Wenxuan Tan
5357f63327
Fix LoRA load from training checkpoint ( #719 )
2025-08-09 20:46:00 -05:00
William Lin
60f61c8101
[bugfix] fix pyproject install and VSA precision test ( #726 )
2025-08-08 18:45:03 -07:00
Wenxuan Tan
6c6bcd914d
Remove all empty_cache ( #713 )
2025-08-07 22:50:38 -07:00
Jinzhe Pan
f2bc037fff
[Fix] training pipeline pin_cpu_memory issue ( #692 )
2025-08-07 02:31:20 -07:00
Jinzhe Pan
86604a684b
[3/3][Preprocess] add preprocessing workflows ( #645 )
2025-08-07 01:49:07 -07:00
Wei Zhou and SolitaryThinker
c41305ad18
[Feat] Add Wan2.2 14B MoE ( #688 )
...
Co-authored-by: SolitaryThinker <wlsaidhi@gmail.com >
2025-08-06 20:31:03 -07:00
William Lin
0ceff110da
[chore] Release 0.1.5 ( #717 )
2025-08-06 13:07:52 -07:00
Yongqi Chen
8d483fe4aa
[Bugfix] Fix neg_prompt bug when training from local cp ( #708 )
2025-08-04 15:54:06 -07:00
Yongqi Chen
c718e56b0d
[Feature] Remove unused args ( #695 )
2025-08-03 23:01:54 -04:00
Yongqi Chen and SolitaryThinker
b65f0316d1
[Feature] Add Wan2.2 DMD example files; Update lr scheduler ( #694 )
...
Co-authored-by: SolitaryThinker <wlsaidhi@gmail.com >
2025-08-03 22:55:01 -04:00
William Lin
8d8bcb76b0
[config] Add config for FastWan2.2 ti2v 5B ( #693 )
2025-08-03 19:09:30 -07:00
Yongqi Chen and SolitaryThinker
5f42748ed1
[Feature] Add Wan2.2-TI2V-5B Sparse Distill ( #690 )
...
Co-authored-by: SolitaryThinker <wlsaidhi@gmail.com >
2025-08-03 01:29:57 -04:00
Yongqi Chen
c9005045dc
[Feature[[Readme] Add VSA/DMD doc ( #673 )
2025-08-02 02:35:46 -04:00
Wenxuan Tan
6c81befc87
[Feature] Optionally enable torch compile ( #684 )
2025-08-01 20:17:40 -07:00
Yongqi Chen
dfe0b288e1
[Bugfix] Add i2v vae loading ( #686 )
2025-08-01 23:15:37 -04:00
Yongqi Chen
9185978c55
[Bugfix] Fix multi-gpu training lr_scheduler ( #682 )
2025-08-01 15:54:03 -04:00
RandNMR73