"cacheflow/core/block_manager.py" did not exist on "c128c2ed0309dd1ff684ac0da2f8fc7ea5994b77"
  • Pi Esposito's avatar
    minimal stable diffusion GPU memory usage with accelerate hooks (#850) · b2e2d141
    Pi Esposito authored
    * add method to enable cuda with minimal gpu usage to stable diffusion
    
    * add test to minimal cuda memory usage
    
    * ensure all models but unet are onn torch.float32
    
    * move to cpu_offload along with minor internal changes to make it work
    
    * make it test against accelerate master branch
    
    * coming back, its official: I don't know how to make it test againt the master branch from accelerate
    
    * make it install accelerate from master on tests
    
    * go back to accelerate>=0.11
    
    * undo prettier formatting on yml files
    
    * undo prettier formatting on yml files againn
    b2e2d141
pr_tests.yml 2.83 KB