Commit graph

389 commits

Author SHA1 Message Date
Frank Lee
ea96cdb5e4 added warmup lr scheduler (#121) 2024-06-05 10:12:52 +08:00
zhengzangw
d71ae85467 [fix] correct video info 2024-06-01 05:08:42 +00:00
zhengzangw
2988d5be99 [fix] read_video for tools 2024-05-31 07:17:01 +00:00
zhengzangw
ba6c800e9a [feat] backend=av 2024-05-30 08:51:22 +00:00
zhengzangw
6fe7aa36b9 Merge branch 'dev/v1.2' of github.com:hpcaitech/Open-Sora-dev into dev/v1.2 2024-05-30 08:49:47 +00:00
zhengzangw
ee1354a7e0 [fix] release av threads 2024-05-30 08:49:39 +00:00
xyupeng
a1a2f29e2b update read_video() 2024-05-30 16:31:37 +08:00
zhengzangw
8a686660bc read video 2024-05-30 08:09:03 +00:00
Zheng Zangwei (Alex Zheng)
9370fbe02d update timer (#115) 2024-05-29 14:46:14 +08:00
Frank Lee
2e1e26a2e4 added timer for benchmarking (#114) 2024-05-29 14:43:15 +08:00
zhengzangw
a50bec768f [feat] complete training with feature 2024-05-21 07:20:14 +00:00
zhengzangw
066b0c9bb3 [wip] feat training 2024-05-21 05:45:06 +00:00
zhengzangw
80f9ecbde3 [feat] generate feature 2024-05-21 04:05:02 +00:00
zhengzangw
c2d499129f Merge branch 'dev/v1.2' of github.com:hpcaitech/Open-Sora-dev into dev/v1.2 2024-05-21 02:38:27 +00:00
Shen Chenhui
d71a06756b Hotfix/vae sampler (#106)
* format

* format

* fix eval loss

* format

* use default seed

* format

* change back ckpt_every to 1k

---------

Co-authored-by: Shen-Chenhui <shen_chenhui@u.nus.edu>
2024-05-21 10:36:26 +08:00
zhengzangw
771100ed88 Merge branch 'dev/v1.2' of github.com:hpcaitech/Open-Sora-dev into dev/v1.2 2024-05-20 09:40:36 +00:00
zhengzangw
bdc67a3933 [wip] debug extract feat 2024-05-20 09:40:28 +00:00
Zheng Zangwei (Alex Zheng)
9cf394884c Dev/load batch (#105)
* add train_load_batch

* update train_load_batch

---------

Co-authored-by: pxy <pexure@gmail.com>
2024-05-20 17:40:05 +08:00
zhengzangw
5b9e753039 [feat] unify dataloader api 2024-05-20 08:40:45 +00:00
zhengzangw
5aad90acaf [feat] extract feature 2024-05-17 09:53:48 +00:00
zhengzangw
a016325a8e [feat] enable freezing y_embedder 2024-05-17 06:40:44 +00:00
zhengzangw
9fd4554f55 [fix] vae dtype 2024-05-16 08:50:24 +00:00
zhengzangw
137d0ac223 [fix] t5 dtype 2024-05-16 08:33:01 +00:00
zhengzangw
0592723022 [feat] update bs search and loss eval 2024-05-15 13:05:53 +00:00
zhengzangw
bb3c23577c [wip] eval loss and search bs 2024-05-15 12:21:07 +00:00
ChangeFWorld
811ac1055b Add Eval loss (#99)
* evaluation half complete

* eval loss finish

* refine config

---------

Co-authored-by: Tianyi <aicfw.li@gmail.com>
2024-05-15 16:17:19 +08:00
zhengzangw
f489cb0915 [feat] accelerate search batch size 2024-05-15 08:07:49 +00:00
zhengzangw
9188b75bc5 Merge branch 'dev/v1.2' of github.com:hpcaitech/Open-Sora-dev into dev/v1.2 2024-05-14 07:25:07 +00:00
zhengzangw
a36f80f847 [feat] notebook for inference 2024-05-14 07:24:56 +00:00
Frank Lee
d1b73a7c9f handled erroneous data (#101) 2024-05-14 14:27:46 +08:00
zhengzangw
e1e379d898 [feat] update inference config 2024-05-14 05:40:17 +00:00
zhengzangw
3a01cb440d [feat] inference for rf 2024-05-13 11:39:16 +00:00
zhengzangw
ff34f2bdaf Merge branch 'dev/v1.2' of github.com:hpcaitech/Open-Sora-dev into dev/v1.2 2024-05-13 06:33:23 +00:00
zhengzangw
6194da6504 [fix] update inference and datautil 2024-05-13 06:33:12 +00:00
Shen Chenhui
b8d99d7a8b fixed (#98)
Co-authored-by: Shen-Chenhui <shen_chenhui@u.nus.edu>
2024-05-13 11:58:06 +08:00
zhengzangw
11faae8db6 [fix] exception handling for dataset 2024-05-12 14:08:37 +00:00
zhengzangw
74dfaa879c [fix] ema ckpt save 2024-05-12 05:15:06 +00:00
zhengzangw
e870139f8f Merge branch 'dev/v1.2' of github.com:hpcaitech/Open-Sora-dev into dev/v1.2 2024-05-10 03:19:48 +00:00
zhengzangw
1db24970d5 updated 2024-05-10 03:19:38 +00:00
Zangwei Zheng
4f27cb51fc Merge branch 'main' into dev/v1.2 2024-05-09 16:49:24 +08:00
zhengzangw
2a55d672a7 [feat] update logging and bucket 2024-05-09 08:07:56 +00:00
Shen Chenhui
89a90e6b3d Hotfix/latent size calc std (#96)
* save progress

* format

* clean config

---------

Co-authored-by: Shen-Chenhui <shen_chenhui@u.nus.edu>
2024-05-09 14:12:29 +08:00
zhengzangw
e9db7b3af1 update training script 2024-05-09 05:53:19 +00:00
zhengzangw
31228fe9c8 [refactor] fix train warnings 2024-05-09 03:10:26 +00:00
zhengzangw
322e566aed update 2024-05-09 03:10:26 +00:00
Zangwei Zheng
eb5ff82fc5 quick update 2024-05-09 11:09:52 +08:00
zhangyongshun
2ec756a240 fix ar defination inconsistency (#375) 2024-05-08 19:48:56 +08:00
Shen Chenhui
dc7b7fd64a fix bug (#95)
Co-authored-by: Shen-Chenhui <shen_chenhui@u.nus.edu>
2024-05-08 17:39:21 +08:00
ChangeFWorld
338bd83f24 mask added, no test due to occupation of gpus (#90)
* mask added, no test due to occupation of gpus

* mask fix

---------

Co-authored-by: ChangeFWorld <aicfw.li@gmail.com>
2024-05-08 16:32:36 +08:00
Zheng Zangwei (Alex Zheng)
f047d5786a [refactor] clean train.py code (#94) 2024-05-08 16:07:57 +08:00