前言¶
本文档介绍 xDiT-GCU 的功能和使用方法,包括模型推理示例和性能评估。
版本信息¶
日期 |
版本 |
新增功能 |
---|---|---|
2024-08-27 |
v 0.2 |
|
原理及功能介绍¶
xDiT-GCU 是适配于燧原 S60 GCU 的 xDiT,用于支持在 Enflame GCU 上基于pytorch native运行 DiT 类模型的推理。
安装使用说明¶
以下步骤基于 Python3.10
, 请先安装所需依赖:
安装环境:安装过程请参考《TopsRider软件栈安装手册》,请根据手册完成TopsRider软件栈安装
安装 torch_gcu
注意:安装 torch_gcu-2.3.0 会自动安装 torch 2.3.0
不同的 Intel/AMD 平台上因为 CPU 架构、指令集的差异,生成的图片可能有细微差异