From a26ee7a1d37a809e1059b83647eec7405bfe5a68 Mon Sep 17 00:00:00 2001 From: Peyton Date: Mon, 7 Aug 2023 16:04:28 -0500 Subject: [PATCH 1/2] Quick fix for queue logic. --- README.md | 2 +- comfyui_to_python.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index ff90f49..058b6a5 100644 --- a/README.md +++ b/README.md @@ -62,7 +62,7 @@ def main(): vaedecodetiled = VAEDecodeTiled() saveimage = SaveImage() - for q in range(1, 10): + for q in range(10): ksampleradvanced_10 = ksampleradvanced.sample( add_noise="enable", noise_seed=random.randint(1, 2**64), diff --git a/comfyui_to_python.py b/comfyui_to_python.py index 3c9d5dd..f4ddfb3 100644 --- a/comfyui_to_python.py +++ b/comfyui_to_python.py @@ -221,7 +221,7 @@ def assemble_python_code(import_statements: set, loader_code: List[str], code: L """ static_imports = ['import random'] imports_code = [f"from nodes import {', '.join([class_name for class_name in import_statements])}" ] - main_function_code = f"def main():\n\t" + '\n\t'.join(loader_code) + f'\n\n\tfor q in {range(1, queue_size)}:\n\t' + '\n\t'.join(code) + main_function_code = f"def main():\n\t" + '\n\t'.join(loader_code) + f'\n\n\tfor q in {range(queue_size)}:\n\t' + '\n\t'.join(code) final_code = '\n'.join(static_imports + ['import sys\nsys.path.append("../")'] + imports_code + ['', main_function_code, '', 'if __name__ == "__main__":', '\tmain()']) final_code = black.format_str(final_code, mode=black.Mode()) From e5254625f857be0cf81b113bb69f881302ac46fc Mon Sep 17 00:00:00 2001 From: Peyton Date: Mon, 7 Aug 2023 16:08:35 -0500 Subject: [PATCH 2/2] changed f string. --- comfyui_to_python.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfyui_to_python.py b/comfyui_to_python.py index f4ddfb3..c31d7b9 100644 --- a/comfyui_to_python.py +++ b/comfyui_to_python.py @@ -221,7 +221,7 @@ def assemble_python_code(import_statements: set, loader_code: List[str], code: L """ static_imports = ['import random'] imports_code = [f"from nodes import {', '.join([class_name for class_name in import_statements])}" ] - main_function_code = f"def main():\n\t" + '\n\t'.join(loader_code) + f'\n\n\tfor q in {range(queue_size)}:\n\t' + '\n\t'.join(code) + main_function_code = f"def main():\n\t" + '\n\t'.join(loader_code) + f'\n\n\tfor q in range({queue_size}):\n\t' + '\n\t'.join(code) final_code = '\n'.join(static_imports + ['import sys\nsys.path.append("../")'] + imports_code + ['', main_function_code, '', 'if __name__ == "__main__":', '\tmain()']) final_code = black.format_str(final_code, mode=black.Mode())