模块化协同设计 (MoCo) 插值器
描述
MoCo 支持抽象的插值器,用于构建和采样各种流行的生成模型框架。具体来说,MoCo 支持连续和离散数据类型的插值器。
连续数据插值器
MoCo 目前支持以下连续数据插值器: - DDPM(去噪扩散概率模型) - VDM(变分扩散模型) - CFM(条件流匹配)
离散数据插值器
MoCo 还支持以下离散数据插值器: - D3PM(离散去噪扩散概率模型) - MDLM(马尔可夫扩散语言模型) - DFM(离散流匹配)
有用的抽象
MoCo 还为可自定义的时间分布和推理时间调度提供了有用的包装器。
可扩展的
如果所需的插值器或采样方法尚未支持,MoCo 被设计为易于扩展。
安装
有关 Conda 环境设置,请参阅 environment
目录中的具体说明。
一旦您的环境设置完成,您可以通过运行以下命令来安装此项目
pip install -e .
示例
请参阅 examples 目录中所有插值器的示例。