Skip to content

Commit

Permalink
[refactor] remove conv_cache from CogVideoX VAE (huggingface#9524)
Browse files Browse the repository at this point in the history
* remove conv cache from the layer and pass as arg instead

* make style

* yiyi's cleaner implementation

Co-Authored-By: YiYi Xu <[email protected]>

* sayak's compiled implementation

Co-Authored-By: Sayak Paul <[email protected]>

---------

Co-authored-by: Sayak Paul <[email protected]>
Co-authored-by: YiYi Xu <[email protected]>
  • Loading branch information
3 people authored Sep 28, 2024
1 parent 1154243 commit bd4df28
Showing 1 changed file with 181 additions and 84 deletions.
Loading

0 comments on commit bd4df28

Please sign in to comment.