mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
e93f4d632d
* introduce new abstraction layer for GPU devices * add unit test for device abstraction * fix ruff * convert TorchDeviceSelect into a stateless class * move logic to select context-specific execution device into context API * add mock hardware environments to pytest * remove dangling mocker fixture * fix unit test for running on non-CUDA systems * remove unimplemented get_execution_device() call * remove autocast precision * Multiple changes: 1. Remove TorchDeviceSelect.get_execution_device(), as well as calls to context.models.get_execution_device(). 2. Rename TorchDeviceSelect to TorchDevice 3. Added back the legacy public API defined in `invocation_api`, including choose_precision(). 4. Added a config file migration script to accommodate removal of precision=autocast. * add deprecation warnings to choose_torch_device() and choose_precision() * fix test crash * remove app_config argument from choose_torch_device() and choose_torch_dtype() --------- Co-authored-by: Lincoln Stein <lstein@gmail.com> |
||
|---|---|---|
| .. | ||
| custom_nodes | ||
| __init__.py | ||
| baseinvocation.py | ||
| collections.py | ||
| compel.py | ||
| constants.py | ||
| controlnet_image_processors.py | ||
| cv.py | ||
| facetools.py | ||
| fields.py | ||
| image.py | ||
| infill.py | ||
| ip_adapter.py | ||
| latent.py | ||
| mask.py | ||
| math.py | ||
| metadata.py | ||
| model.py | ||
| noise.py | ||
| param_easing.py | ||
| primitives.py | ||
| prompt.py | ||
| sdxl.py | ||
| strings.py | ||
| t2i_adapter.py | ||
| tiles.py | ||
| upscale.py | ||
| util.py | ||