环境:
操作系统:统信V20
Docker image: image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
git clone http://developer.hpccube.com/codes/modelzoo/flux.1_diffusers.git
cd flux.1_diffusers
git submodule init && git submodule update
python Flux_test.py
使用xformers计算attention:
export FLUX_USE_XFORMERS=1
python Flux_test.py
之前弄错了,现在可以了。
暂无数据
暂无数据