From b58cb3c189d18ebbdcf654374f7adaa844f946dc Mon Sep 17 00:00:00 2001 From: Peyton DeNiro Date: Sat, 19 Aug 2023 21:05:28 -0500 Subject: [PATCH] Update README.md --- README.md | 102 ++++++++++++------------------------------------------ 1 file changed, 22 insertions(+), 80 deletions(-) diff --git a/README.md b/README.md index 4032123..5afbcf3 100644 --- a/README.md +++ b/README.md @@ -10,74 +10,18 @@ The `ComfyUI-to-Python-Extension` is a powerful tool that translates ComfyUI wor **To this:** ``` -import os import random -import sys -from typing import Sequence, Mapping, Any, Union import torch +import sys - -def add_comfyui_directory_to_sys_path() -> None: - """ - Recursively looks at parent folders starting from the current working directory until it finds 'ComfyUI'. - Once found, the directory is added to sys.path. - """ - start_path = os.getcwd() # Get the current working directory - - def search_directory(path: str) -> None: - # Check if the current directory contains 'ComfyUI' - if "ComfyUI" in os.listdir(path): - directory_path = os.path.join(path, "ComfyUI") - sys.path.append(directory_path) - print(f"ComfyUI found and added to sys.path: {directory_path}") - - # Get the parent directory - parent_directory = os.path.dirname(path) - - # If the parent directory is the same as the current directory, we've reached the root and stop the search - if parent_directory == path: - return - - # Recursively call the function with the parent directory - search_directory(parent_directory) - - # Start the search from the current working directory - search_directory(start_path) - - -def get_value_at_index(obj: Union[Sequence, Mapping], index: int) -> Any: - """Returns the value at the given index of a sequence or mapping. - - If the object is a sequence (like list or string), returns the value at the given index. - If the object is a mapping (like a dictionary), returns the value at the index-th key. - - Some return a dictionary, in these cases, we look for the "results" key - - Args: - obj (Union[Sequence, Mapping]): The object to retrieve the value from. - index (int): The index of the value to retrieve. - - Returns: - Any: The value at the given index. - - Raises: - IndexError: If the index is out of bounds for the object and the object is not a mapping. - """ - try: - return obj[index] - except KeyError: - return obj["result"][index] - - -add_comfyui_directory_to_sys_path() +sys.path.append("../") from nodes import ( - CLIPTextEncode, - KSamplerAdvanced, - CheckpointLoaderSimple, VAEDecode, - SaveImage, + KSamplerAdvanced, EmptyLatentImage, - NODE_CLASS_MAPPINGS, + SaveImage, + CheckpointLoaderSimple, + CLIPTextEncode, ) @@ -95,12 +39,12 @@ def main(): cliptextencode = CLIPTextEncode() cliptextencode_6 = cliptextencode.encode( - text="Kylo Ren trapped inside of a Mark Rothko painting", - clip=get_value_at_index(checkpointloadersimple_4, 1), + text="evening sunset scenery blue sky nature, glass bottle with a galaxy in it", + clip=checkpointloadersimple_4[1], ) cliptextencode_7 = cliptextencode.encode( - text="text, watermark", clip=get_value_at_index(checkpointloadersimple_4, 1) + text="text, watermark", clip=checkpointloadersimple_4[1] ) checkpointloadersimple_12 = checkpointloadersimple.load_checkpoint( @@ -108,13 +52,12 @@ def main(): ) cliptextencode_15 = cliptextencode.encode( - text="Kylo Ren trapped inside of a Mark Rothko painting", - clip=get_value_at_index(checkpointloadersimple_12, 1), + text="evening sunset scenery blue sky nature, glass bottle with a galaxy in it", + clip=checkpointloadersimple_12[1], ) cliptextencode_16 = cliptextencode.encode( - text="text, watermark", - clip=get_value_at_index(checkpointloadersimple_12, 1), + text="text, watermark", clip=checkpointloadersimple_12[1] ) ksampleradvanced = KSamplerAdvanced() @@ -132,10 +75,10 @@ def main(): start_at_step=0, end_at_step=20, return_with_leftover_noise="enable", - model=get_value_at_index(checkpointloadersimple_4, 0), - positive=get_value_at_index(cliptextencode_6, 0), - negative=get_value_at_index(cliptextencode_7, 0), - latent_image=get_value_at_index(emptylatentimage_5, 0), + model=checkpointloadersimple_4[0], + positive=cliptextencode_6[0], + negative=cliptextencode_7[0], + latent_image=emptylatentimage_5[0], ) ksampleradvanced_11 = ksampleradvanced.sample( @@ -148,19 +91,18 @@ def main(): start_at_step=20, end_at_step=10000, return_with_leftover_noise="disable", - model=get_value_at_index(checkpointloadersimple_12, 0), - positive=get_value_at_index(cliptextencode_15, 0), - negative=get_value_at_index(cliptextencode_16, 0), - latent_image=get_value_at_index(ksampleradvanced_10, 0), + model=checkpointloadersimple_12[0], + positive=cliptextencode_15[0], + negative=cliptextencode_16[0], + latent_image=ksampleradvanced_10[0], ) vaedecode_17 = vaedecode.decode( - samples=get_value_at_index(ksampleradvanced_11, 0), - vae=get_value_at_index(checkpointloadersimple_12, 2), + samples=ksampleradvanced_11[0], vae=checkpointloadersimple_12[2] ) saveimage_19 = saveimage.save_images( - filename_prefix="ComfyUI", images=get_value_at_index(vaedecode_17, 0) + filename_prefix="ComfyUI", images=vaedecode_17[0] )