Skip to content

GitLab

Commit 6305e94b authored by dengjb's avatar dengjb
Browse files

update

parent 4e6ad5f7
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
README.md
View file @ 6305e94b
...@@ -32,7 +32,7 @@ Z-Image 是一个强大的、高效的图像生成模型，具有 6B 参数。 ...@@ -32,7 +32,7 @@ Z-Image 是一个强大的、高效的图像生成模型，具有 6B 参数。
| flash_attn | 2.6.1+das.opt1.dtk2504 | | flash_attn | 2.6.1+das.opt1.dtk2504 |
| flash_mla | 1.0.0+das.opt1.dtk25042 | | flash_mla | 1.0.0+das.opt1.dtk25042 |
当前仅支持镜像: 推荐使用镜像:
- 挂载地址`-v`根据实际模型情况修改 - 挂载地址`-v`根据实际模型情况修改
```bash ```bash
...@@ -52,8 +52,9 @@ docker run -it --shm-size 60g --network=host --name z-image --privileged --devic ...@@ -52,8 +52,9 @@ docker run -it --shm-size 60g --network=host --name z-image --privileged --devic
### pytorch ### pytorch
#### 单机推理 #### 单机推理
可参考run.sh脚本
```python ```python
# 可参考run.sh脚本
import torch import torch
from diffusers import ZImagePipeline from diffusers import ZImagePipeline
...@@ -105,7 +106,7 @@ DCU与GPU精度一致，推理框架：pytorch。 ...@@ -105,7 +106,7 @@ DCU与GPU精度一致，推理框架：pytorch。
## 预训练权重 ## 预训练权重
| 模型名称 | 权重大小 | DCU型号 | 最低卡数需求 |下载地址| | 模型名称 | 权重大小 | DCU型号 | 最低卡数需求 |下载地址|
|:-----:|:----------:|:----------:|:---------------------:|:----------:| |:-----:|:----------:|:----------:|:---------------------:|:----------:|
| Z-Image-Turbo | 6B | BW1000 | 1 | [下载地址](https://modelscope.cn/models/Tongyi-MAI/Z-Image-Turbo) | | Z-Image-Turbo | 6B | BW1000 | 1 | [modelscope](https://modelscope.cn/models/Tongyi-MAI/Z-Image-Turbo) |
## 源码仓库及问题反馈 ## 源码仓库及问题反馈
- https://developer.sourcefind.cn/codes/modelzoo/z-image_pytorch - https://developer.sourcefind.cn/codes/modelzoo/z-image_pytorch
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment