mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
18 lines
522 B
Python
18 lines
522 B
Python
"""
|
|
Initialization file for invokeai.backend.util
|
|
"""
|
|
from .attention import auto_detect_slice_size # noqa: F401
|
|
from .devices import ( # noqa: F401
|
|
CPU_DEVICE,
|
|
CUDA_DEVICE,
|
|
MPS_DEVICE,
|
|
choose_precision,
|
|
choose_torch_device,
|
|
normalize_device,
|
|
torch_dtype,
|
|
)
|
|
from .logging import InvokeAILogger
|
|
from .util import Chdir, ask_user, download_with_resume, instantiate_from_config, url_attachment_name # noqa: F401
|
|
|
|
__all__ = ["Chdir", "InvokeAILogger", "choose_precision", "choose_torch_device"]
|