ComfyUI to Python Extension

Comfyui-to-Python-Extension

A ComfyUI-to-Python-Extension é uma ferramenta poderosa que traduz fluxos de trabalho conforfyui em código Python executável. Projetado para preencher a lacuna entre a interface visual da Comfyui e o ambiente de programação da Python, esse script facilita a transição perfeita do design para a execução do código. Seja você um cientista de dados, um desenvolvedor de software ou um entusiasta da IA, essa ferramenta otimiza o processo de implementação de fluxos de trabalho confortáveis no Python.

Converta isso:

Para isso:

 import random
import torch
import sys

sys.path.append("../")
from nodes import (
    VAEDecode,
    KSamplerAdvanced,
    EmptyLatentImage,
    SaveImage,
    CheckpointLoaderSimple,
    CLIPTextEncode,
)


def main():
    with torch.inference_mode():
        checkpointloadersimple = CheckpointLoaderSimple()
        checkpointloadersimple_4 = checkpointloadersimple.load_checkpoint(
            ckpt_name="sd_xl_base_1.0.safetensors"
        )

        emptylatentimage = EmptyLatentImage()
        emptylatentimage_5 = emptylatentimage.generate(
            width=1024, height=1024, batch_size=1
        )

        cliptextencode = CLIPTextEncode()
        cliptextencode_6 = cliptextencode.encode(
            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=checkpointloadersimple_4[1]
        )

        checkpointloadersimple_12 = checkpointloadersimple.load_checkpoint(
            ckpt_name="sd_xl_refiner_1.0.safetensors"
        )

        cliptextencode_15 = cliptextencode.encode(
            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=checkpointloadersimple_12[1]
        )

        ksampleradvanced = KSamplerAdvanced()
        vaedecode = VAEDecode()
        saveimage = SaveImage()

        for q in range(10):
            ksampleradvanced_10 = ksampleradvanced.sample(
                add_noise="enable",
                noise_seed=random.randint(1, 2**64),
                steps=25,
                cfg=8,
                sampler_name="euler",
                scheduler="normal",
                start_at_step=0,
                end_at_step=20,
                return_with_leftover_noise="enable",
                model=checkpointloadersimple_4[0],
                positive=cliptextencode_6[0],
                negative=cliptextencode_7[0],
                latent_image=emptylatentimage_5[0],
            )

            ksampleradvanced_11 = ksampleradvanced.sample(
                add_noise="disable",
                noise_seed=random.randint(1, 2**64),
                steps=25,
                cfg=8,
                sampler_name="euler",
                scheduler="normal",
                start_at_step=20,
                end_at_step=10000,
                return_with_leftover_noise="disable",
                model=checkpointloadersimple_12[0],
                positive=cliptextencode_15[0],
                negative=cliptextencode_16[0],
                latent_image=ksampleradvanced_10[0],
            )

            vaedecode_17 = vaedecode.decode(
                samples=ksampleradvanced_11[0], vae=checkpointloadersimple_12[2]
            )

            saveimage_19 = saveimage.save_images(
                filename_prefix="ComfyUI", images=vaedecode_17[0]
            )


if __name__ == "__main__":
    main()

Casos de uso potencial

• Simplificando o processo para criar um aplicativo enxuto ou implantação de pipeline que use um fluxo de trabalho confortável
• Criando experimentos programáticos para vários valores de prompt/parâmetros
• Criando grandes filas para geração de imagens (por exemplo, você pode ajustar o script para gerar 1000 imagens sem clicar em Ctrl+digite 1000 vezes)
• Expandir -se ou iterando facilmente em sua arquitetura em Python, uma vez que um fluxo de trabalho fundamental está em vigor na GUI

V1.3.0 Notas de liberação

• Gerar arquivo .py diretamente do aplicativo Web Comfyui

V1.2.1 Notas de liberação

• Altere dinamicamente os parâmetros comfyui_to_python.py com os parâmetros com argumentos da CLI
• Hotfix para lidar com nós que aceitam Kwargs.

V1.2.0 Notas de liberação

• Atualizações para aderir às últimas alterações de ComfyUI

V1.0.0 Notas de liberação

• Use todos os nós personalizados!
  • Os nós personalizados agora são suportados. Se você tiver algum problema com a execução do código, primeiro verifique se cada nó funciona conforme o esperado na GUI. Se funcionar na GUI, mas não no script gerado, envie um problema.

Instalação

1. Navegue até o seu diretório ComfyUI/custom_nodes

2. Clone este repo

   git clone https://github.com/pydn/ComfyUI-to-Python-Extension.git

   Depois de clonar o repositório, seu diretório ComfyUI deve ficar assim:

    /comfy
   /comfy_extras
   /custom_nodes
   --/ComfyUI-to-Python-Extension
   /input
   /models
   /output
   /script_examples
   /web
   .gitignore
   LICENSE
   README.md
   comfyui_screenshot.png
   cuda_mollac.py
   execution.py
   extra_model_paths.yaml.example
   folder_paths.py
   latent_preview.py
   main.py
   nodes.py
   requirements.txt
   server.py
   

Uso do aplicativo da web

1. Inicie confortyui

2. Carregue seu fluxo de trabalho favorito e clique em Save As Script

3. Digite o nome do arquivo desejado na tela pop -up.

4. Mova o arquivo .py da sua pasta de downloads para o seu diretório ComfyUI .

5. Agora você pode executar o arquivo .py recém -criado para gerar imagens sem iniciar um servidor.

Uso da CLI

1. Navegue até a pasta ComfyUI-to-Python-Extension e instale

   pip install -r requirements.txt

2. Inicie a Comfyui, clique no ícone de engrenagem sobre Queue Prompt e verifique Enable Dev mode Options . O script não funcionará se você não habilitar esta opção!

3. Carregue seus fluxos de trabalho favoritos e clique no botão Save (API Format) recém -habilitado em Pump de fila

4. Mova o arquivo de fluxo de trabalho .json baixado para sua pasta ComfyUI/ComfyUI-to-Python-Extension

5. Se necessário, adicione argumentos ao executar comfyui_to_python.py para atualizar o input_file padrão e output_file para corresponder ao seu arquivo de fluxo de trabalho .json e nome de arquivo .py desejado. Por padrão, o script procurará um arquivo chamado workflow_api.json . Você também pode atualizar a variável queue_size para o número desejado de imagens que deseja gerar em uma única execução de script. Por padrão, os scripts gerarão 10 imagens. Execute python comfyui_to_python.py --help para obter mais detalhes.

6a. Execute o script com argumentos padrão:

python comfyui_to_python.py

6b. Execute o script com argumentos opcionais:

python comfyui_to_python.py --input_file " workflow_api (2).json " --output_file my_workflow.py --queue_size 100

7. Depois de executar comfyui_to_python.py , um novo arquivo .py será criado no diretório de trabalho atual. Se você não fez alterações, procure workflow_api.py .

8. Agora você pode executar o arquivo .py recém -criado para gerar imagens sem iniciar um servidor.