• Patrick von Platen's avatar
    Refactor model offload (#4514) · 93579650
    Patrick von Platen authored
    
    
    * [Draft] Refactor model offload
    
    * [Draft] Refactor model offload
    
    * Apply suggestions from code review
    
    * cpu offlaod updates
    
    * remove model cpu offload from individual pipelines
    
    * add hook to offload models to cpu
    
    * clean up
    
    * model offload
    
    * add model cpu offload string
    
    * make style
    
    * clean up
    
    * fixes for offload issues
    
    * fix tests issues
    
    * resolve merge conflicts
    
    * update src/diffusers/pipelines/pipeline_utils.py
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    
    * make style
    
    * Update src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py
    
    ---------
    Co-authored-by: default avatarDhruv Nair <dhruv.nair@gmail.com>
    93579650
pipeline_utils.py 87 KB