2023-03-03 06:02:00 +00:00
|
|
|
"""
|
2023-03-03 05:02:15 +00:00
|
|
|
Initialization file for invokeai.backend.util
|
2023-03-03 06:02:00 +00:00
|
|
|
"""
|
2023-09-11 13:57:41 +00:00
|
|
|
from .attention import auto_detect_slice_size # noqa: F401
|
2023-08-17 22:45:25 +00:00
|
|
|
from .devices import ( # noqa: F401
|
2023-03-03 06:02:00 +00:00
|
|
|
CPU_DEVICE,
|
|
|
|
|
CUDA_DEVICE,
|
|
|
|
|
MPS_DEVICE,
|
|
|
|
|
choose_precision,
|
|
|
|
|
choose_torch_device,
|
|
|
|
|
normalize_device,
|
|
|
|
|
torch_dtype,
|
|
|
|
|
)
|
2023-11-25 00:37:46 +00:00
|
|
|
from .logging import InvokeAILogger
|
2023-11-26 18:18:21 +00:00
|
|
|
from .util import Chdir, ask_user, download_with_resume, instantiate_from_config, url_attachment_name # noqa: F401
|
2023-11-25 00:37:46 +00:00
|
|
|
|
2023-11-26 22:13:31 +00:00
|
|
|
__all__ = ["Chdir", "InvokeAILogger", "choose_precision", "choose_torch_device"]
|
