bulk update styling and formatting (#170)

finetrainers/utils/__init__.py

@@ -1,5 +1,5 @@
-from .file_utils import find_files, delete_files
-from .diffusion_utils import resolution_dependant_timestep_flow_shift, default_flow_shift
-from .memory_utils import get_memory_statistics, bytes_to_gigabytes, free_memory, make_contiguous
-from .torch_utils import unwrap_model
+from .diffusion_utils import default_flow_shift, resolution_dependant_timestep_flow_shift
+from .file_utils import delete_files, find_files
+from .memory_utils import bytes_to_gigabytes, free_memory, get_memory_statistics, make_contiguous
 from .optimizer_utils import get_optimizer, gradient_norm, max_gradient
+from .torch_utils import unwrap_model

finetrainers/utils/checkpointing.py

@@ -1,8 +1,11 @@
 import os
 from typing import Tuple
+
 from accelerate.logging import get_logger
+
 from ..constants import FINETRAINERS_LOG_LEVEL
-from ..utils.file_utils import find_files, delete_files
+from ..utils.file_utils import delete_files, find_files
+
 
 logger = get_logger("finetrainers")
 logger.setLevel(FINETRAINERS_LOG_LEVEL)

finetrainers/utils/data_utils.py

@@ -3,7 +3,7 @@ from typing import Union
 
 from accelerate.logging import get_logger
 
-from ..constants import PRECOMPUTED_DIR_NAME, PRECOMPUTED_CONDITIONS_DIR_NAME, PRECOMPUTED_LATENTS_DIR_NAME
+from ..constants import PRECOMPUTED_CONDITIONS_DIR_NAME, PRECOMPUTED_DIR_NAME, PRECOMPUTED_LATENTS_DIR_NAME
 
 
 logger = get_logger("finetrainers")

finetrainers/utils/file_utils.py

@@ -2,7 +2,7 @@ import logging
 import os
 import shutil
 from pathlib import Path
-from typing import Any, Dict, List, Union
+from typing import List, Union
 
 
 logger = logging.getLogger("finetrainers")

finetrainers/utils/memory_utils.py

@@ -1,5 +1,4 @@
 import gc
-import logging
 from typing import Any, Dict, Union
 
 import torch

finetrainers/utils/optimizer_utils.py

@@ -1,7 +1,8 @@
 import inspect
 
-from accelerate.logging import get_logger
 import torch
+from accelerate.logging import get_logger
+
 
 logger = get_logger("finetrainers")
 
@@ -45,7 +46,7 @@ def get_optimizer(
 
     if (use_torchao and (use_8bit or use_4bit)) or use_cpu_offload_optimizer:
         try:
-            import torchao
+            import torchao  # noqa
 
         except ImportError:
             raise ImportError(

finetrainers/utils/torch_utils.py

@@ -32,4 +32,4 @@ def align_device_and_dtype(
 def expand_tensor_to_dims(tensor, ndim):
     while len(tensor.shape) < ndim:
         tensor = tensor.unsqueeze(-1)
-    return tensor
+    return tensor