nodes/pydn-ComfyUI-to-Python-Extension
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Quick fix for queue logic.

Browse Source
This commit is contained in:
Peyton
2023-08-07 16:04:28 -05:00
parent 255370cd82
commit a26ee7a1d3
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+1 -1
View File
@@ -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),
comfyui_to_python.py
+1 -1
View File
@@ -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())