Comfyui_image2prompt 项目使用教程
1. 项目的目录结构及介绍
Comfyui_image2prompt/
├── custom_nodes/
│ └── Comfyui_image2prompt/
│ ├── __init__.py
│ ├── main.py
│ ├── config.yaml
│ └── models/
│ ├── moondream1/
│ ├── moondream2/
│ └── ...
├── README.md
└── requirements.txt
custom_nodes/
: 包含插件的主要代码和配置文件。Comfyui_image2prompt/
: 插件的核心目录。__init__.py
: 初始化文件。main.py
: 项目的启动文件。config.yaml
: 项目的配置文件。models/
: 存放模型文件的目录。moondream1/
: moondream1 模型文件。moondream2/
: moondream2 模型文件。- ...
README.md
: 项目说明文档。requirements.txt
: 项目依赖文件。
2. 项目的启动文件介绍
main.py
是项目的启动文件,负责初始化插件并启动 ComfyUI 环境。以下是 main.py
的主要功能:
import os
import sys
from custom_nodes.Comfyui_image2prompt import config
def main():
# 读取配置文件
config.load_config()
# 初始化插件
init_plugin()
# 启动 ComfyUI 环境
start_comfyui()
if __name__ == "__main__":
main()
config.load_config()
: 读取并加载配置文件config.yaml
。init_plugin()
: 初始化插件,包括加载模型等。start_comfyui()
: 启动 ComfyUI 环境,使插件生效。
3. 项目的配置文件介绍
config.yaml
是项目的配置文件,用于配置插件的各种参数。以下是 config.yaml
的一个示例:
plugin:
name: Comfyui_image2prompt
version: 1.0.0
models:
- name: moondream1
path: models/moondream1
- name: moondream2
path: models/moondream2
settings:
max_prompt_length: 100
enable_auto_download: true
plugin
: 插件的基本信息。name
: 插件名称。version
: 插件版本。models
: 插件使用的模型列表。name
: 模型名称。path
: 模型路径。
settings
: 插件的运行参数。max_prompt_length
: 生成的提示语最大长度。enable_auto_download
: 是否自动下载模型。
通过配置文件,用户可以自定义插件的行为和使用的模型,从而灵活地调整插件的功能。