Formatting

Author: ionite34

install.py

@@ -8,14 +8,13 @@
 
 def pip_install(package: str) -> None:
     subprocess.check_call(
-        [sys.executable, "-m", "pip", "install", package],
-        cwd=PARENT_DIR
+        [sys.executable, "-m", "pip", "install", package], cwd=PARENT_DIR
     )
 
 
 def main() -> None:
     pip_install("-e .")
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()

src/inference_core_nodes/prompt_expansion/prompt_expansion.py

@@ -16,10 +16,7 @@
 
 CONFIGS_DIR = Path(__file__).parent.joinpath("configs")
 
-fooocus_magic_split = [
-    ", extremely",
-    ", intricate,"
-]
+fooocus_magic_split = [", extremely", ", intricate,"]
 
 disallowed_chars_table = str.maketrans("", "", "[]【】()（）|:：")
 
@@ -46,7 +43,9 @@ def __init__(self, model_directory: str):
         self.tokenizer = AutoTokenizer.from_pretrained(model_directory)
         self.model = AutoModelForCausalLM.from_pretrained(model_directory)
 
-        positive_tokens = model_directory.joinpath("positive.txt").read_text().splitlines()
+        positive_tokens = (
+            model_directory.joinpath("positive.txt").read_text().splitlines()
+        )
 
         positive_tokens = []
 
@@ -86,7 +85,7 @@ def __call__(self, prompt: str, seed: int) -> str:
         )
 
         response = self.tokenizer.batch_decode(features, skip_special_tokens=True)
-        result = response[0][len(origin):]
+        result = response[0][len(origin) :]
         result = safe_str(result)
         result = result.translate(disallowed_chars_table)
         return result
@@ -110,7 +109,7 @@ def INPUT_TYPES(s):
             "required": {
                 "model_name": (folder_paths.get_filename_list("prompt_expansion"),),
                 "text": ("STRING", {"multiline": True}),
-                "seed": ("INT", {"default": 0, "min": 0, "max": 0xFFFFFFFF})
+                "seed": ("INT", {"default": 0, "min": 0, "max": 0xFFFFFFFF}),
             },
         }
 
@@ -151,4 +150,6 @@ def expand_prompt(model_name: str, text: str, seed: int, log_prompt: str):
 NODE_CLASS_MAPPINGS = {"Inference_Core_PromptExpansion": PromptExpansion}
 
 # A dictionary that contains human-readable titles for the nodes
-NODE_DISPLAY_NAME_MAPPINGS = {"Inference_Core_PromptExpansion": "[Inference.Core] Prompt Expansion"}
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "Inference_Core_PromptExpansion": "[Inference.Core] Prompt Expansion"
+}

src/inference_core_nodes/prompt_expansion/util.py

@@ -11,4 +11,4 @@ def join_prompts(*args, **kwargs):
         return ""
     if len(prompts) == 1:
         return prompts[0]
-    return ', '.join(prompts)
+    return ", ".join(prompts)