mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
532f82cb97
* avoid copying model back from cuda to cpu * handle models that don't have state dicts * add assertions that models need a `device()` method * do not rely on torch.nn.Module having the device() method * apply all patches after model is on the execution device * fix model patching in latents too * log patched tokenizer * closes #6375 --------- Co-authored-by: Lincoln Stein <lstein@gmail.com> |
||
---|---|---|
.. | ||
load | ||
metadata | ||
util | ||
__init__.py | ||
config.py | ||
convert_ckpt_to_diffusers.py | ||
libc_util.py | ||
merge.py | ||
probe.py | ||
search.py | ||
starter_models.py |