• Arsalan's avatar
    Device agnostic testing (#5612) · f427345a
    Arsalan authored
    * utils and test modifications to enable device agnostic testing
    
    * device for manual seed in unet1d
    
    * fix generator condition in vae test
    
    * consistency changes to testing
    
    * make style
    
    * add device agnostic testing changes to source and one model test
    
    * make dtype check fns private, log cuda fp16 case
    
    * remove dtype checks from import utils, move to testing_utils
    
    * adding tests for most model classes and one pipeline
    
    * fix vae import
    f427345a
testing_utils.py 33.2 KB