• Sayak Paul's avatar
    [tests] refactor vae tests (#9808) · c1926cef
    Sayak Paul authored
    
    
    * add: autoencoderkl tests
    
    * autoencodertiny.
    
    * fix
    
    * asymmetric autoencoder.
    
    * more
    
    * integration tests for stable audio decoder.
    
    * consistency decoder vae tests
    
    * remove grad check from consistency decoder.
    
    * cog
    
    * bye test_models_vae.py
    
    * fix
    
    * fix
    
    * remove allegro
    
    * fixes
    
    * fixes
    
    * fixes
    
    ---------
    Co-authored-by: default avatarDhruv Nair <dhruv.nair@gmail.com>
    c1926cef
autoencoder_tiny.py 15.7 KB