Lincoln Stein e93f4d632d [util] Add generic torch device class (#6174) ... * 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>		2024-04-15 13:12:49 +00:00
..
ip_adapter	Create a UNetAttentionPatcher for patching UNet models with CustomAttnProcessor2_0 modules.	2024-04-09 08:12:12 -04:00
model_manager	after stopping install and download services, wait for thread exit	2024-03-21 16:41:20 +11:00
tiles	In CalculateImageTilesEvenSplitInvocation to have overlap_fraction becomes just overlap. This is now in pixels rather than as a fraction of the tile size.	2023-12-17 15:10:50 +00:00
util	[util] Add generic torch device class (#6174)	2024-04-15 13:12:49 +00:00
__init__.py	POC of a test that depends on models.	2023-10-05 15:35:58 -04:00