前言

本文档介绍 xDiT-GCU 的功能和使用方法,包括模型推理示例和性能评估。

版本信息

表 13 版本信息

日期

版本

新增功能

2024-08-27

v 0.2

在 gcu 上支持了下述模型或 pipeline 的推理:
  • stable-diffusion-xl-base-1.0;

  • HunyuanDiT-v1.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 架构、指令集的差异,生成的图片可能有细微差异