InvokeAI/ldm/invoke/generator
Kevin Turner 8a0d45ac5a
new OffloadingDevice loads one model at a time, on demand (#2596)
* new OffloadingDevice loads one model at a time, on demand

* fixup! new OffloadingDevice loads one model at a time, on demand

* fix(prompt_to_embeddings): call the text encoder directly instead of its forward method

allowing any associated hooks to run with it.

* more attempts to get things on the right device from the offloader

* more attempts to get things on the right device from the offloader

* make offloading methods an explicit part of the pipeline interface

* inlining some calls where device is only used once

* ensure model group is ready after pipeline.to is called

* fixup! Strategize slicing based on free [V]RAM (#2572)

* doc(offloading): docstrings for offloading.ModelGroup

* doc(offloading): docstrings for offloading-related pipeline methods

* refactor(offloading): s/SimpleModelGroup/FullyLoadedModelGroup

* refactor(offloading): s/HotSeatModelGroup/LazilyLoadedModelGroup

to frame it is the same terms as "FullyLoadedModelGroup"

---------

Co-authored-by: Damian Stewart <null@damianstewart.com>
2023-02-16 23:48:27 +00:00
..
__init__.py rename all modules from ldm.dream to ldm.invoke 2022-10-08 11:37:23 -04:00
base.py Fix perlin noise generator for diffusers tensors (#2678) 2023-02-15 19:37:42 +01:00
diffusers_pipeline.py new OffloadingDevice loads one model at a time, on demand (#2596) 2023-02-16 23:48:27 +00:00
embiggen.py use 🧨diffusers model (#1583) 2023-01-15 09:22:46 -05:00
img2img.py Add thresholding for all diffusers types (#2479) 2023-02-14 18:00:34 -06:00
inpaint.py fix crash when text mask applied to img2img (#2526) 2023-02-05 15:26:40 +00:00
omnibus.py use 🧨diffusers model (#1583) 2023-01-15 09:22:46 -05:00
txt2img2img.py Add thresholding for all diffusers types (#2479) 2023-02-14 18:00:34 -06:00
txt2img.py Add thresholding for all diffusers types (#2479) 2023-02-14 18:00:34 -06:00