Commit graph

94 commits

Author SHA1 Message Date
Zangwei Zheng
3944e93065 update training config 2024-04-22 12:05:59 +08:00
zhengzangw
6073af6444 merge video edit 2024-04-20 15:47:48 +00:00
zhengzangw
d262f14a24 update 2024-04-20 02:43:38 +00:00
Zangwei Zheng
15f9d702ed [exp] update 2024-04-19 17:43:44 +08:00
Zangwei Zheng
dce2ef4a1c [exp] new config 2024-04-19 15:21:00 +08:00
Zheng Zangwei (Alex Zheng)
d2a782efac Dev/fps (#55)
* support fps

* update fps
2024-04-19 13:18:52 +08:00
Zheng Zangwei (Alex Zheng)
069ea0d687 Feat/fast bucket (#54)
* [wip] bucket

* [bug] not parallel

* update eval

* update sample.sh

* accelerate bucket build with pandarallel
2024-04-19 11:42:02 +08:00
Zheng Zangwei (Alex Zheng)
1f99b9ba4b complete eval pipeline (#53) 2024-04-18 15:49:14 +08:00
FrankLeeeee
6ee754bef9 print error data path 2024-04-17 16:45:18 +08:00
Zangwei Zheng
3b85effe5a a bunch of update for data 2024-04-13 15:44:24 +08:00
Zangwei Zheng
fff002bede [feat] support parquet 2024-04-11 15:28:49 +08:00
Zangwei Zheng
e1d98678be [fix] dataset ext judge 2024-04-08 20:44:01 +08:00
Zangwei Zheng
8d3f0f8282 [fix] dataset ext judge 2024-04-08 20:42:38 +08:00
Zangwei Zheng
50c0a35658 [fix] bucket 2024-04-08 02:28:14 +08:00
Zangwei Zheng
2685f2930e fix bucket index 2024-04-07 19:48:44 +08:00
Zangwei Zheng
02b6335628 fix bucket index 2024-04-07 19:35:30 +08:00
Zangwei Zheng
882225897b fix bucket index 2024-04-07 16:28:44 +08:00
Zheng Zangwei (Alex Zheng)
7612d22fc6 Exp/image mix (#20)
* [exp] image mixed training

* [exp] add batch info

* [exp] launch

* update num_step_per_epoch

* [feat] verify image mix training
2024-04-02 13:35:09 +08:00
Frank Lee
baf91f835e updated sampler workload and state dict (#18) 2024-04-01 21:10:51 +08:00
Frank Lee
b8b78c6733 Hotfix/sampler (#16)
* fixed variable video sampler

* polish

* polish
2024-04-01 20:37:41 +08:00
Zangwei Zheng
c553ee274f better aspect ratio 2024-04-01 14:27:13 +08:00
Frank Lee
a9e8220c7c fixed image size in sampler (#15)
* fixed image size in sampler

* polish
2024-04-01 11:43:56 +08:00
Frank Lee
46f9af34fc fixed multiprocessing dataloader (#14) 2024-03-31 20:59:22 +08:00
Zangwei Zheng
987283fa1b [fix] transform may not fit enough 2024-03-30 17:05:15 +08:00
Zangwei Zheng
52aeb3769f format and some fix 2024-03-30 12:02:48 +08:00
Hongxin Liu
5e82d1493b [feature] refactor sampler (#4)
* [feature] refactor sampler

* [feature] support sampler resuming

* [feature] support sampler resuming
2024-03-29 23:32:12 +08:00
Zangwei Zheng
2d4a5df287 [wip] image wrong with flashattn 2024-03-28 22:04:43 +08:00
Zangwei Zheng
a01f6da20e [feat] support for stdit2 sampling 2024-03-28 21:35:33 +08:00
Zangwei Zheng
cba02e8a58 [feat] dynamic for video (base_size not completed) 2024-03-28 17:58:36 +08:00
Zangwei Zheng
94e686177e [wip] multi ar 2024-03-27 23:04:12 +08:00
Zangwei Zheng
7392d2e551 [feat] multiple frames with 360p 2024-03-27 00:24:46 +08:00
Zangwei Zheng
62097da2d3 refactor temporal_random_crop 2024-03-26 17:32:15 +08:00
Zangwei Zheng
552b7e8f79 register dataset 2024-03-26 17:02:41 +08:00
Zangwei Zheng
01728dc28a refactor datasets 2024-03-26 16:50:36 +08:00
Zangwei Zheng
f3ace2b341 merge stdit2 2024-03-26 15:46:31 +08:00
Zangwei Zheng
13423c57a7 train dataset support new format 2024-03-25 18:36:56 +08:00
Zangwei Zheng
98e62a7c57 update inference z 2024-03-23 20:28:34 +08:00
Zangwei Zheng
7d27f5553e merge mask-related utils 2024-03-23 16:32:51 +08:00
Frank Lee
6b2715e40c Feature/gradio demo (#190)
* [gradio] added demo app

* polish
2024-03-22 15:07:04 +08:00
Frank Lee
9d8188744f added licenses (#80) 2024-03-17 21:59:37 +08:00
Zangwei Zheng
921e3138b3 update docs 2024-03-17 15:47:48 +08:00
Zheng Zangwei (Alex Zheng)
150cf4666a Docs/readme (#74)
* update docs

* update docs

* update docs

* update acceleration docs and fix typos
2024-03-16 21:17:16 +08:00
Zheng Zangwei (Alex Zheng)
d851a85535 format (#69) 2024-03-15 22:16:20 +08:00
xyupeng
9aab3ad343 added open_sora package (#66) 2024-03-15 22:00:46 +08:00