Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
lwh
660e09d2e9
|
1 year ago | |
|---|---|---|
| configs | 1 year ago | |
| demo | 1 year ago | |
| ldm | 1 year ago | |
| README.md | 1 year ago | |
| README_EN.md | 1 year ago | |
| infer.sh | 1 year ago | |
| requirements.txt | 1 year ago | |
| txt2img.py | 1 year ago | |
README.md
悟空画画
目录
Wukong-Huahua悟空画画模型
Wukong-Huahua是基于扩散模型的中文文生图大模型，由华为诺亚团队携手中软分布式并行实验室，昇腾计算产品部联合开发。模型基于Wukong dataset训练，并使用昇思框架(MindSpore)+昇腾(Ascend)软硬件解决方案实现。
欢迎访问我们的在线体验平台试玩。
环境依赖
-
昇腾软硬件解决方案(驱动+固件+CANN)
前往昇腾社区，按照说明下载安装。
-
AI框架 - MindSpore == 1.9
前往MindSpore官网，按照说明下载安装。
如需更多帮助，可以参考以下资料
-
第三方依赖
pip install -r requirements.txt
快速开始
准备checkpoint
下载Wukong-Huahua预训练参数 wukong-huahua-ms.ckpt 至 wukong-huahua/models/ 目录.
文图生成
要进行文图生成，可以运行txt2img.py 或者直接使用默认参数运行 infer.sh.
python txt2img.py --prompt [input text] --ckpt_path [ckpt_path] --ckpt_name [ckpt_name] \
--H [image_height] --W [image_width] --output_path [image save folder] \
--n_samples [number of images to generate]
或者
bash infer.sh
更高的分辨率需要更大的显存. 对于 Ascend 910 芯片, 我们可以同时生成2张1024x768的图片或者16张512x512的图片。
生成样例
下面是我们的Wukong-Huahua模型生成的一些样例以及对应的[input text]。
城市夜景 赛博朋克 格雷格·鲁特科夫斯基
莫奈 撑阳伞的女人 月亮 梦幻
海上日出时候的奔跑者
诺亚方舟在世界末日起航 科幻插画
时空 黑洞 辐射
乡村 田野 屏保
来自深渊 风景 绘画 写实风格
