• Pi Esposito's avatar
    add accelerate to load models with smaller memory footprint (#361) · 4d1cce2f
    Pi Esposito authored
    
    
    * add accelerate to load models with smaller memory footprint
    
    * remove low_cpu_mem_usage as it is reduntant
    
    * move accelerate init weights context to modelling utils
    
    * add test to ensure results are the same when loading with accelerate
    
    * add tests to ensure ram usage gets lower when using accelerate
    
    * move accelerate logic to single snippet under modelling utils and remove it from configuration utils
    
    * format code using to pass quality check
    
    * fix imports with isor
    
    * add accelerate to test extra deps
    
    * only import accelerate if device_map is set to auto
    
    * move accelerate availability check to diffusers import utils
    
    * format code
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    4d1cce2f
import_utils.py 9.75 KB