mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
093174942b
`generator` now asks `InvokeAIDiffuserComponent` to do postprocessing work on latents after every step. Thresholding - now implemented as replacing latents outside of the threshold with random noise - is called at this point. This postprocessing step is also where we can hook up symmetry and other image latent manipulations in the future. Note: code at this layer doesn't need to worry about MPS as relevant torch functions are wrapped and made MPS-safe by `generator.py`. |
||
---|---|---|
.. | ||
__init__.py | ||
classifier.py | ||
cross_attention_control.py | ||
cross_attention_map_saving.py | ||
ddim.py | ||
ddpm.py | ||
ksampler.py | ||
plms.py | ||
sampler.py | ||
shared_invokeai_diffusion.py |