| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
Support stage-1 CP training, SANA-WM Benchmark & fp8/fp4 quantized streaming inference (#404) * feat(sana-wm): support stage-1 CP training * docs(sana-wm): expose chunk-causal training entry * style: run pre-commit on sana-wm training changes * docs: add SANA-WM benchmark guide * SANA-WM streaming: per-component fp8/fp4 quantized inference Add --stage1_precision/--refiner_precision (bf16/fp8/fp4) with per-block fp8_autocast scoping that preserves action-following, fp8 recipe selector for stage-1 + refiner, default Transformer Engine install, and benchmark docs. * Polish Sana-WM PR docs and formatting * Use end-to-end Sana-WM inference CI * Add SANA-WM stage1 training CI * Run SANA-WM stage1 smoke in video CI * Keep SANA-WM training in dedicated CI job --------- Co-authored-by: junsong <cjs1020440147@icloud.com> | 3 个月前 | |
SANA Streaming Training Code (#421) * bidirectional training release * release sana streaming training code * Fix pre-commit formatting * test: cover streaming paired data and chunks * fix typo, remove unused code, add re-try boundary * pre-commit * test: cover bidirectional streaming training * fix: bound V2V batch retries locally * fix reverse generation overlap * add online validation * test: cover reverse overlap and V2V validation --------- Co-authored-by: junsong <cjs1020440147@icloud.com> | 2 个月前 | |
Update DC-AE-Lite (#299) * change all `dc-ae-turbo` to `dc-ae-lite` * update CI test; * fix missing file; * code update; * fix bugs; | 1 年前 | |
[SANA-Video] Adding SANA-Video Training & Inference code & Dataloader and so on (#313) * 1. support sana-video training; 2. support wan model; 3. add dataloader for video training; 4. add all we need to train a SANA-Video; * update train scripts code; * 1. update train scripts code; 2. update wan code; * remove used code; * update config files * ci code update; train code update; * add inference code and run success; * remove unused code; * fix path error; * update several readme for sana-video * [CI-Lint] Fix code style issues with pre-commit 7bc87de3966810a7ff9741da009dfe22f1154cb4 * update readme; Co-authored-by: Enze Xie <johnny_ez@163.com> * update version * fix bug; * update env Co-authored-by: xieenze <Johnny_ez@163.com> * env update; * code update; * change env; * env update; * update env; * env update; * update toy data download script; * update code; * skip ci env test; * update env; * update; * fix * pre-commit; * fix; * pre-commit; * fix; * fix; * remove python setup; * fix; * pre-commit * fix bug; * fix bug; * fix bug; * fix bug; * fix bug; * remove flex attn; * fix train bug; * seperate data downloading in ci; * pre-commit * update; * pre-commit * update; * fix * fix no gpu import bug; fix test train video bug; * pre-commit * update env in each ci test; * fix; * code update; * update; * Update Sana_2000M_256px_AdamW_fsdp.yaml * Update Sana_2000M_480px_AdamW_fsdp.yaml * fix * ivjoint is not supported in ci; * fix; * Remove HunyuanVAE and duplicated siglip * Delete diffusion/model/dc_ae/efficientvit/models copy/__init__.py * Remove old ar sampler * Self-forcing and longlive will have a new scheduler * Add ChunkFlowEuler back, in case someone want to train AR * Remove siglip * Delete diffusion/scheduler/ar_flow_euler_sampler.py * remove peft load * remove duplicated wan config * [CI-Lint] Fix code style issues with pre-commit ec5160791010e0f820e900ba6ceeb73ad9d8529d * Remove chunk ar long video scheduler and self-forcing scheduler * Remove task=df * Update builder.py * Update config.py * [CI-Lint] Fix code style issues with pre-commit 0f3d3b6d935499792dae6267917c3f57536b7377 * fix config bug * [CI-Lint] Fix code style issues with pre-commit baa4215169f13bd549038fe0557b104f68014527 --------- Co-authored-by: GitHub Action <action@github.com> Co-authored-by: Enze Xie <johnny_ez@163.com> Co-authored-by: Yuyang Zhao <43061147+HeliosZhao@users.noreply.github.com> Co-authored-by: HeliosZhao <yuyangzhao@outlook.com> | 11 个月前 |