• Aryan's avatar
    [training] CogVideoX Lora (#9302) · 2b443a5d
    Aryan authored
    
    
    * 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: default avatarYiYi Xu <yixu310@gmail.com>
    
    * fix lora alpha
    
    * use correct lora scaling for final test pipeline
    
    * Update examples/cogvideo/train_cogvideox_lora.py
    Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
    
    * apply suggestions from review; prodigy optimizer
    
    YiYi Xu <yixu310@gmail.com>
    
    * 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: default avatarFu-Yun Wang <1697256461@qq.com>
    Co-Authored-By: default avatarzR <2448370773@qq.com>
    
    * fight lora runner tests
    
    * import Dummy optim and scheduler only wheh required
    
    * update docs
    
    * add coauthors
    Co-Authored-By: default avatarFu-Yun Wang <1697256461@qq.com>
    
    * remove option to train text encoder
    Co-Authored-By: default avatarbghira <bghira@users.github.com>
    
    * 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 f8a8444487db27859be812866db4e8cec7f25691.
    
    * 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: default avatarYiYi Xu <yixu310@gmail.com>
    
    ---------
    Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
    Co-authored-by: default avatarzR <2448370773@qq.com>
    Co-authored-by: default avatarFu-Yun Wang <1697256461@qq.com>
    Co-authored-by: default avatarbghira <bghira@users.github.com>
    2b443a5d
__init__.py 3.99 KB