Commit graph

437 commits

Author SHA1 Message Date
FrankLeeeee
6a72b8910b [data] added error handling to dataset 2024-06-24 08:53:10 +00:00
zhengzangw
7115864314 [fix] HF loading 2024-06-22 15:41:32 +00:00
zhengzangw
cd12584034 handle av error 2024-06-22 13:26:55 +00:00
zhengzangw
a6bdabe286 minor fix 2024-06-21 19:22:02 +00:00
zhengzangw
7aa940f20d Merge branch 'main' of https://github.com/hpcaitech/Open-Sora into dev/v1.2 2024-06-21 19:17:30 +00:00
zhengzangw
20d1584a1c [fix] support stdit1 training 2024-06-21 19:03:30 +00:00
zhengzangw
dec17bd990 [feat] reduce memory leakage in dataloader and pyav 2024-06-21 18:23:30 +00:00
Zheng Zangwei (Alex Zheng)
9b668e1c4e Merge pull request #523 from BurkeHulk/hotfix/fp16_nan_output
Force fp16 input to fp32 to avoid nan output in timestep_transform
2024-06-21 18:01:17 +08:00
HangXu
04d2ee0182 Force fp16 input to fp32 to avoid nan output in timestep_transform 2024-06-21 11:15:39 +03:00
zhengzangw
f32c1173b7 config for local load 2024-06-20 10:23:38 +00:00
rangoliu
81524e675e fix ar keys (#500) 2024-06-20 17:51:24 +08:00
Shen Chenhui
416837a86b Hotfix/vae (#502)
* fix assert

* fix vae config; update path

---------

Co-authored-by: Shen-Chenhui <shen_chenhui@u.nus.edu>
2024-06-20 17:49:16 +08:00
Zheng Zangwei (Alex Zheng)
4cbf3c33b8 Hotfix/t5 load (#487)
* hotfix

* hotfix for stdit

* hotfix for vae
2024-06-19 23:15:29 +08:00
Zheng Zangwei (Alex Zheng)
396307c050 Hotfix/t5 load (#486)
* hotfix

* hotfix for stdit
2024-06-19 23:03:15 +08:00
Zheng Zangwei (Alex Zheng)
85f20274a0 hotfix (#484) 2024-06-19 22:47:40 +08:00
Zheng Zangwei (Alex Zheng)
ccb85fc3c3 hotfix (#482) 2024-06-19 22:05:47 +08:00
Zheng Zangwei (Alex Zheng)
403772eee1 Docs/fix zangwei (#474)
* [docs] fix training data num

* [docs] update sp

* add support for issue #470
2024-06-19 16:53:53 +08:00
Jianshu Guo
17cce908b2 fix:multi-fps bug. For multi-fps training, when extracting frames according to a certain frame_interval, the fps of the extracted frames actually changes. (#444)
Co-authored-by: Jianshu Guo <fguojianshu@gmail.com>
2024-06-18 16:20:11 +08:00
jackmappotion
2229f01b35 ♻️ Directory creation to use os.makedirs with exist_ok (#435)
Simplified code and improved readability / Ensured functionality remains the same by allowing directory to exist without error
2024-06-18 07:41:02 +08:00
FrankLeeeee
7ba29d3439 [doc] resolved conflict in readme 2024-06-17 23:18:40 +00:00
zhengzangw
525e29abbe reformat and update docs 2024-06-17 15:37:23 +00:00
zhengzangw
30e276166c update 2024-06-17 13:42:27 +00:00
Shen Chenhui
f0c98dd186 Merge pull request #148 from hpcaitech/feature/docs_v1.2
Feature/docs v1.2
2024-06-17 17:18:27 +08:00
Shen-Chenhui
d2e710c977 format 2024-06-17 09:13:12 +00:00
zhengzangw
dc577e98b9 Merge branch 'dev/v1.2' of github.com:hpcaitech/Open-Sora-dev into dev/v1.2 2024-06-17 07:01:35 +00:00
zhengzangw
41f772d1f8 [docs] update readme 2024-06-17 07:01:27 +00:00
Frank Lee
4d243955ca [checkpoint] adapted vae to hf (#145) 2024-06-17 14:59:33 +08:00
zhengzangw
b6b2535c79 update inference 2024-06-17 05:57:25 +00:00
Frank Lee
93ba7719ec [checkpoint] adapted inference to hf pretrained weights (#143) 2024-06-17 11:27:24 +08:00
Frank Lee
be61f44a29 [gradio] udpated inference logic (#138)
* [gradio] udpated inference logic

* polish
2024-06-15 22:43:33 +08:00
zhengzangw
b70b749e55 [fix] loop support for score and refinement 2024-06-14 17:18:29 +00:00
Frank Lee
41e6e9492e [gradio] supported random prompt and watermark (#136) 2024-06-14 17:57:44 +08:00
zhengzangw
c5e0475749 [feat] support for random sample 2024-06-14 08:50:40 +00:00
zhengzangw
c93e846017 [feat] fps control 2024-06-14 07:44:35 +00:00
zhengzangw
992aae7d8f [feat] prompt refine 2024-06-14 07:37:00 +00:00
zhengzangw
9117c9fcc7 [docs] update 2024-06-14 03:34:06 +00:00
Frank Lee
a887f54711 added sp for stdit3 (#131) 2024-06-14 11:10:43 +08:00
Frank Lee
9dde960b95 added gradio (#133)
* added gradio

* added gradio
2024-06-13 16:56:30 +08:00
zhengzangw
a8a34d56b9 [fix] inference with score 2024-06-13 02:37:45 +00:00
zhengzangw
51cdd377cc reformat 2024-06-12 09:24:53 +00:00
Shen-Chenhui
b16d6f5ebe format 2024-06-12 08:26:31 +00:00
zhengzangw
43fa324a6e [wip] human evaluation 2024-06-12 06:57:30 +00:00
zhengzangw
a271faa131 [fix] condition sampling 2024-06-11 07:49:00 +00:00
Shen-Chenhui
bd182d93dd format 2024-06-10 06:02:36 +00:00
zhengzangw
6094f5fe92 [feat] add score for inference 2024-06-09 12:01:26 +00:00
zhengzangw
ce601066c5 [stage2] update config 2024-06-05 15:49:51 +00:00
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