• Aryan's avatar
    [refactor] CogVideoX followups + tiled decoding support (#9150) · a85b34e7
    Aryan authored
    * refactor context parallel cache; update torch compile time benchmark
    
    * add tiling support
    
    * make style
    
    * remove num_frames % 8 == 0 requirement
    
    * update default num_frames to original value
    
    * add explanations + refactor
    
    * update torch compile example
    
    * update docs
    
    * update
    
    * clean up if-statements
    
    * address review comments
    
    * add test for vae tiling
    
    * update docs
    
    * update docs
    
    * update docstrings
    
    * add modeling test for cogvideox transformer
    
    * make style
    a85b34e7
autoencoder_kl_cogvideox.py 50.6 KB