ComfyUI-BAGEL 的安装和配置教程

项目基础介绍

ComfyUI-BAGEL 是一个基于 ComfyUI 的自定义节点包，依托于 BAGEL-7B-MoT 多模态模型开发而成。该项目旨在提供文本到图像生成、图像编辑、图像理解等功能，适用于创意设计和内容生成等领域。

该项目主要使用 Python 编程语言。

项目使用的关键技术和框架

ComfyUI-BAGEL 使用了以下关键技术和框架：

• BAGEL（Bridge Across modalities with Generative Encoder and Language）: 这是一个开源的多模态基础模型，具有 7B 活性参数（总共 14B），采用了混合变压器专家（MoT）架构。
• ComfyUI: 一个用于创建和训练自定义节点和模型的用户界面。

项目安装和配置的准备工作

在开始安装 ComfyUI-BAGEL 之前，请确保您的系统已满足以下要求：

• Python 环境已安装，推荐使用 Python 3.7 或更高版本。
• Git 已安装并配置好。
• GPU 硬件支持（对于图像生成等任务，建议使用具备较高显存容量的 GPU）。

安装步骤

以下是安装 ComfyUI-BAGEL 的详细步骤：

1. 下载模型

   当首次使用 ComfyUI-BAGEL 时，BAGEL-7B-MoT 模型将自动下载至 models/bagel/BAGEL-7B-MoT/ 目录下。您也可以手动下载模型：

   # 使用 git lfs 克隆模型（推荐）
   git lfs install
   git clone https://huggingface.co/ByteDance-Seed/BAGEL-7B-MoT models/bagel/BAGEL-7B-MoT
   

   或者使用 Huggingface_hub：

   pip install huggingface_hub
   python -c "
   from huggingface_hub import snapshot_download;
   snapshot_download(repo_id='ByteDance-Seed/BAGEL-7B-MoT', local_dir='models/bagel/BAGEL-7B-MoT')
   "
   

2. 安装依赖

   在项目目录下执行以下命令安装所需的依赖：

   pip install -r requirements.txt
   

3. 重启 ComfyUI

   为了加载新的节点，需要重启 ComfyUI。

按照上述步骤操作后，您就可以开始使用 ComfyUI-BAGEL 进行多模态理解和生成任务了。