loaders for main, controlnet, ip-adapter, clipvision and t2i

invokeai/backend/util/__init__.py

@@ -12,14 +12,22 @@ from .devices import (  # noqa: F401
     torch_dtype,
 )
 from .logging import InvokeAILogger
-from .util import ( # TO DO: Clean this up; remove the unused symbols
+from .util import (  # TO DO: Clean this up; remove the unused symbols
     GIG,
     Chdir,
     ask_user,  # noqa
     directory_size,
     download_with_resume,
-    instantiate_from_config, # noqa
+    instantiate_from_config,  # noqa
     url_attachment_name,  # noqa
-    )
+)
 
-__all__ = ["GIG", "directory_size","Chdir", "download_with_resume", "InvokeAILogger", "choose_precision", "choose_torch_device"]
+__all__ = [
+    "GIG",
+    "directory_size",
+    "Chdir",
+    "download_with_resume",
+    "InvokeAILogger",
+    "choose_precision",
+    "choose_torch_device",
+]

invokeai/backend/util/devices.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from contextlib import nullcontext
-from typing import Union, Optional
+from typing import Optional, Union
 
 import torch
 from torch import autocast

invokeai/backend/util/util.py

@@ -27,6 +27,7 @@ from .devices import torch_dtype
 # actual size of a gig
 GIG = 1073741824
 
+
 def directory_size(directory: Path) -> int:
     """
     Return the aggregate size of all files in a directory (bytes).
@@ -39,6 +40,7 @@ def directory_size(directory: Path) -> int:
             sum += Path(root, d).stat().st_size
     return sum
 
+
 def log_txt_as_img(wh, xc, size=10):
     # wh a tuple of (width, height)
     # xc a list of captions to plot