Fixes

1 files changed, 11 insertions, 5 deletions

diff --git a/util/files.py b/util/files.py
index 2712525..73ff802 100644
--- a/util/files.py
+++ b/util/files.py
@@ -8,7 +8,7 @@ from safetensors import safe_open
 
 
 def load_config(filename):
-    with open(filename, 'rt') as f:
+    with open(filename, "rt") as f:
         config = json.load(f)
 
     args = config["args"]
@@ -19,11 +19,17 @@ def load_config(filename):
     return args
 
 
-def load_embeddings_from_dir(tokenizer: MultiCLIPTokenizer, embeddings: ManagedCLIPTextEmbeddings, embeddings_dir: Path):
+def load_embeddings_from_dir(
+    tokenizer: MultiCLIPTokenizer,
+    embeddings: ManagedCLIPTextEmbeddings,
+    embeddings_dir: Path,
+):
     if not embeddings_dir.exists() or not embeddings_dir.is_dir():
-        return []
+        return [], []
 
-    filenames = [filename for filename in embeddings_dir.iterdir() if filename.is_file()]
+    filenames = [
+        filename for filename in embeddings_dir.iterdir() if filename.is_file()
+    ]
     tokens = [filename.stem for filename in filenames]
 
     new_ids: list[list[int]] = []
@@ -39,7 +45,7 @@ def load_embeddings_from_dir(tokenizer: MultiCLIPTokenizer, embeddings: ManagedC
 
     embeddings.resize(len(tokenizer))
 
-    for (new_id, embeds) in zip(new_ids, new_embeds):
+    for new_id, embeds in zip(new_ids, new_embeds):
         embeddings.add_embed(new_id, embeds)
 
     return tokens, new_ids