混元DiT Github頁(yè)面

為了提升開(kāi)發(fā)者使用體驗(yàn)，騰訊混元官方上線了專屬的加速庫(kù)，讓推理時(shí)間縮短75%，提高大模型運(yùn)行效率。開(kāi)發(fā)者通過(guò)Hugging Face即可下載該推理加速工具。

項(xiàng)目組通過(guò)知識(shí)蒸餾和TensorRT高性能推理框架，實(shí)現(xiàn)了DiT模型的采樣步數(shù)壓縮與高效推理部署。蒸餾主要指降低擴(kuò)散模型迭代的步數(shù)實(shí)現(xiàn)加速。模型整體結(jié)構(gòu)和參數(shù)量不變，用戶在無(wú)需任何額外的操作和設(shè)備要求下使用蒸餾權(quán)重，即可降低50%迭代步數(shù)，實(shí)現(xiàn)耗時(shí)減半。TensorRT推理加速方案通過(guò)工程優(yōu)化，可以進(jìn)一步降低耗時(shí)，并且和模型權(quán)重解耦。同時(shí)使用兩者進(jìn)行推理部署，可將推理時(shí)間縮短75%。

官方還分享了兩個(gè)降低用戶使用門(mén)檻的最新消息：經(jīng)過(guò)與吐司社區(qū)的共同努力，用戶可以基于ComfyUI的圖形化界面使用騰訊混元文生圖模型能力。同時(shí)，通過(guò)與Hugging Face團(tuán)隊(duì)合作，混元DiT模型已經(jīng)部署到Hugging Face官方模型庫(kù)Diffusers中，并編寫(xiě)了適配該模型庫(kù)的調(diào)用和生成代碼，用戶可以直接通過(guò)該途徑調(diào)用混元DiT模型，大大簡(jiǎn)化了用戶使用的成本。

ComfyUI是一款文生圖領(lǐng)域的WebUI界面設(shè)計(jì)，它將文生圖領(lǐng)域的擴(kuò)散算法模塊化與圖形化，提升了生成效率與資源利用率，也大幅降低了開(kāi)發(fā)者的使用門(mén)檻。用戶可以通過(guò)圖像化工作流使用混元DiT文生圖模型，實(shí)現(xiàn)與官方模型一樣的效果。

混元DiT文生圖模型的ComfyUI使用界面

此外，圍繞ComfyUI的使用生態(tài)，還衍生出強(qiáng)大的開(kāi)源社區(qū)?；煸狣iT對(duì)ComfyUI的支持，也能讓社區(qū)的成員體驗(yàn)基于最新DiT架構(gòu)的文生圖模型。

作為知名的AI開(kāi)源社區(qū)，Hugging Face的Diffusers是目前調(diào)用各種主流文生圖大模型的通用庫(kù)，已經(jīng)成為當(dāng)今文生圖大模型使用的社區(qū)標(biāo)準(zhǔn)。

將混元DiT模型適配進(jìn)Hugging Face Diffusers，可以大幅提升模型的易用性和用戶基礎(chǔ)。用戶無(wú)需將原始代碼庫(kù)下載與部署到自身環(huán)境，安裝了Diffusers庫(kù)的開(kāi)發(fā)者，僅需要運(yùn)行幾行代碼，即可調(diào)用混元DiT模型，配置與調(diào)用都十分方便。同時(shí)，Hugging Face與騰訊混元團(tuán)隊(duì)共同優(yōu)化算法框架，加快了圖片的生成速度。

此舉也相當(dāng)于為所有后續(xù)基于混元DiT的使用和開(kāi)發(fā)提供了底層支持，覆蓋各種需要調(diào)用混元DiT的任何場(chǎng)景，包括上述的ComfyUI方式。同時(shí)對(duì)開(kāi)發(fā)者而言，之前配置好的基于Diffusers的工作流和插件可以在少量修改的情況下直接用于混元DiT。

騰訊文生圖負(fù)責(zé)人蘆清林表示：“騰訊混元文生圖模型開(kāi)源之后得到了眾多開(kāi)發(fā)者的支持和反饋，我們十分高興，也同時(shí)在針對(duì)開(kāi)發(fā)者的反饋與社區(qū)一起完善和優(yōu)化基于混元DiT的開(kāi)源生態(tài)，讓更多開(kāi)發(fā)者能更便利地享受到最新的研究成果。也歡迎大家跟我們一起共建下一代視覺(jué)生成開(kāi)源生態(tài)，推動(dòng)大模型行業(yè)加速發(fā)展?！?/p>

崔歡歡