minddiffusion/vision/Taichu-GLIDE

• Taichu-GLIDE
• • 模型介绍
  • 环境要求
  • 快速体验
  • • 推理
    • 训练

Taichu-GLIDE

模型介绍

Taichu-GLIDE是华为昇腾计算携手武汉人工智能研究院、中科院自动化所基于昇腾昇思全栈开发的中文文生图大模型（紫东.太初系列模型之一），该模型采用了AIGC领域当前非常流行的扩散模型（Diffusion Model）技术，代码和预训练模型权重均对外进行开源，开发者可使用本仓进行以文生图任务的体验。

                                                                                                 一幅画着柯基的油画

环境要求

1. 安装 CANN（5.1.RC2 版本）及其配套版本的驱动（driver）和 固件（firemware）
   前往昇腾社区下载安装包：
   https://www.hiascend.com/software/cann/commercial
   以arm + 欧拉的系统配置为例 (x86的系统请选择x86的包)

2. 安装 MindSpore 1.8.1 版本
   前往MindSpore官网，按照教程安装对应版本，链接如下:
   https://www.mindspore.cn/install

3. 安装 requirements 依赖
   pip install -r requirements.txt

快速体验

推理

• 请先点击此处下载ckpt文件
• 在data/prompts.txt添加自己想要生成的prompt
• 修改 scripts/run_infer.sh中相关路径及配置

bash scripts/run_infer.sh

训练

# 生成阶段分布式训练
bash scripts/run_gen_finetune_dist.sh [DEVICE_NUM] [VISIABLE_DEVICES(0,1,2,3,4,5,6,7)] [RANK_TABLE_FILE]

# 超分阶段分布式训练
bash scripts/run_super_res_finetune_dist.sh [DEVICE_NUM] [VISIABLE_DEVICES(0,1,2,3,4,5,6,7)] [RANK_TABLE_FILE]