本文主要是介绍T-GATE 无需要训练加速diffusion模型; PixArt-Alpha LCM再加速使用,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
参考:https://github.com/HaozheLiu-ST/T-GATE
git clone https://github.com/HaozheLiu-ST/T-GATE.git
transformers 4.38
torch 2.1.2+cu118
代码:
inference_step 4-10高点效果会好点,这步数比原来20来步的LCM确实也有提升
python main.py --prompt 'a dog' --model 'lcm_pixart' --gate_step 1
这篇关于T-GATE 无需要训练加速diffusion模型; PixArt-Alpha LCM再加速使用的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!