跳到内容

模块化协同设计 (MoCo) 插值器

描述

MoCo 支持抽象的插值器,用于构建和采样各种流行的生成模型框架。具体来说,MoCo 支持连续和离散数据类型的插值器。

连续数据插值器

MoCo 目前支持以下连续数据插值器: - DDPM(去噪扩散概率模型) - VDM(变分扩散模型) - CFM(条件流匹配)

离散数据插值器

MoCo 还支持以下离散数据插值器: - D3PM(离散去噪扩散概率模型) - MDLM(马尔可夫扩散语言模型) - DFM(离散流匹配)

有用的抽象

MoCo 还为可自定义的时间分布和推理时间调度提供了有用的包装器。

可扩展的

如果所需的插值器或采样方法尚未支持,MoCo 被设计为易于扩展。

安装

有关 Conda 环境设置,请参阅 environment 目录中的具体说明。

一旦您的环境设置完成,您可以通过运行以下命令来安装此项目

pip install -e .
这将以可编辑模式安装项目,允许您进行更改并立即看到反映。

示例

请参阅 examples 目录中所有插值器的示例。