ComfyUI_TensorRT 项目使用指南

ComfyUI_TensorRT 项目使用指南

ComfyUI_TensorRT项目地址:https://gitcode.com/gh_mirrors/co/ComfyUI_TensorRT

1. 项目的目录结构及介绍

ComfyUI_TensorRT/
├── workflows/
│   ├── example_workflow1.json
│   ├── example_workflow2.json
│   └── ...
├── tensorrt/
│   ├── model1_engine.trt
│   ├── model2_engine.trt
│   └── ...
├── README.md
├── LICENSE
├── requirements.txt
└── ...

• workflows/: 包含项目的不同工作流程配置文件，每个 .json 文件定义了一个具体的工作流程。
• tensorrt/: 存放生成的 TensorRT 引擎文件，每个 .trt 文件对应一个优化后的模型引擎。
• README.md: 项目的基本介绍和使用说明。
• LICENSE: 项目的开源许可证文件。
• requirements.txt: 项目依赖的 Python 包列表。

2. 项目的启动文件介绍

项目的主要启动文件是 ComfyUI.py，该文件负责初始化项目环境、加载配置和启动主界面。启动项目的基本命令如下：

python ComfyUI.py

3. 项目的配置文件介绍

项目的配置文件主要位于 config/ 目录下，包含以下几个关键配置文件：

• config.yaml: 主配置文件，包含项目的全局设置，如模型路径、日志级别等。
• workflows.yaml: 工作流程配置文件，定义了不同工作流程的具体参数和步骤。
• tensorrt_config.yaml: TensorRT 引擎的配置文件，包含引擎生成和加载的相关参数。

配置文件采用 YAML 格式，易于阅读和编辑。以下是一个示例配置片段：

model_path: "tensorrt/model1_engine.trt"
log_level: "INFO"
workflows:
  - name: "Example Workflow 1"
    steps:
      - load_checkpoint: "checkpoint1.pth"
      - convert_to_tensorrt: "model1"

通过编辑这些配置文件，用户可以自定义项目的运行行为和参数。

ComfyUI_TensorRT项目地址:https://gitcode.com/gh_mirrors/co/ComfyUI_TensorRT