参考:https://github.com/gojasper/flash-diffusion
https://huggingface.co/jasperai/flash-pixart
安装包注意:
diffusers 这里是官方上面有更改,参考:https://github.com/gojasper/flash-diffusion/blob/main/requirements.txt
pip install -r requirements.txt
diffusers @ git+https://github.com/initml/diffusers.git@clement/feature/flash
--extra-index-url https://download.pytorch.org/whl/cu118