Skip to content

Commit fd15a9d

Browse files
a-r-r-o-wyiyixuxuzRzRzRzRzRzRzRG-U-Nbghira
authored andcommitted
[training] CogVideoX Lora (#9302)
* cogvideox lora training draft * update * update * update * update * update * make fix-copies * update * update * apply suggestions from review * apply suggestions from reveiw * fix typo * Update examples/cogvideo/train_cogvideox_lora.py Co-authored-by: YiYi Xu <[email protected]> * fix lora alpha * use correct lora scaling for final test pipeline * Update examples/cogvideo/train_cogvideox_lora.py Co-authored-by: YiYi Xu <[email protected]> * apply suggestions from review; prodigy optimizer YiYi Xu <[email protected]> * add tests * make style * add README * update * update * make style * fix * update * add test skeleton * revert lora utils changes * add cleaner modifications to lora testing utils * update lora tests * deepspeed stuff * add requirements.txt * deepspeed refactor * add lora stuff to img2vid pipeline to fix tests * fight tests * add co-authors Co-Authored-By: Fu-Yun Wang <[email protected]> Co-Authored-By: zR <[email protected]> * fight lora runner tests * import Dummy optim and scheduler only wheh required * update docs * add coauthors Co-Authored-By: Fu-Yun Wang <[email protected]> * remove option to train text encoder Co-Authored-By: bghira <[email protected]> * update tests * fight more tests * update * fix vid2vid * fix typo * remove lora tests; todo in follow-up PR * undo img2vid changes * remove text encoder related changes in lora loader mixin * Revert "remove text encoder related changes in lora loader mixin" This reverts commit f8a8444. * update * round 1 of fighting tests * round 2 of fighting tests * fix copied from comment * fix typo in lora test * update styling Co-Authored-By: YiYi Xu <[email protected]> --------- Co-authored-by: YiYi Xu <[email protected]> Co-authored-by: zR <[email protected]> Co-authored-by: Fu-Yun Wang <[email protected]> Co-authored-by: bghira <[email protected]>
1 parent 240f753 commit fd15a9d

File tree

11 files changed

+2618
-195
lines changed

11 files changed

+2618
-195
lines changed

examples/cogvideo/README.md

Lines changed: 228 additions & 0 deletions
Large diffs are not rendered by default.

examples/cogvideo/requirements.txt

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
accelerate>=0.31.0
2+
torchvision
3+
transformers>=4.41.2
4+
ftfy
5+
tensorboard
6+
Jinja2
7+
peft>=0.11.1
8+
sentencepiece
9+
decord>=0.6.0
10+
imageio-ffmpeg

0 commit comments

Comments
 (0)