騰訊混元文生圖開(kāi)源模型推理再加速,可支持ComfyUI圖形化工作流

6月6日,騰訊發(fā)布針對(duì)騰訊混元文生圖開(kāi)源大模型(以下簡(jiǎn)稱(chēng)為混元DiT模型)的加速庫(kù),讓推理效率大幅提升,生圖時(shí)間縮短75%。

混元DiT模型的使用門(mén)檻也大幅降低。用戶(hù)可以基于ComfyUI的圖形化界面,使用騰訊混元文生圖模型能力。同時(shí),混元DiT模型已經(jīng)部署至Hugging Face Diffusers通用模型庫(kù)中,用戶(hù)僅用三行代碼即可調(diào)用混元DiT模型,無(wú)需下載原始代碼庫(kù)。

此前,騰訊宣布旗下的混元文生圖大模型全面升級(jí)并對(duì)外開(kāi)源,可供企業(yè)與個(gè)人開(kāi)發(fā)者免費(fèi)商用。這是業(yè)內(nèi)首個(gè)中文原生的DiT架構(gòu)文生圖開(kāi)源模型,支持中英文雙語(yǔ)輸入及理解;采用了與 sora 一致的DiT架構(gòu),不僅可支持文生圖,也可作為視頻等多模態(tài)視覺(jué)生成的基礎(chǔ)。

騰訊混元DiT模型開(kāi)源之后,受到了眾多社區(qū)開(kāi)發(fā)者的認(rèn)可。開(kāi)源不到一個(gè)月,項(xiàng)目Github Star數(shù)就超過(guò)2100,位于開(kāi)源社區(qū)熱門(mén)DiT模型前列。

為了提升開(kāi)發(fā)者使用體驗(yàn),騰訊混元官方上線了專(zhuā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ù)量不變,用戶(hù)在無(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è)降低用戶(hù)使用門(mén)檻的最新消息:經(jīng)過(guò)與社區(qū)的共同努力,用戶(hù)可以基于ComfyUI的圖形化界面使用騰訊混元文生圖模型能力。同時(shí),通過(guò)與Hugging Face團(tuán)隊(duì)合作,混元DiT模型已經(jīng)部署到Hugging Face官方模型庫(kù)Diffusers中,并編寫(xiě)了適配該模型庫(kù)的調(diào)用和生成代碼,用戶(hù)可以直接通過(guò)該途徑調(diào)用混元DiT模型,大大簡(jiǎn)化了用戶(hù)使用的成本。

ComfyUI是一款文生圖領(lǐng)域的WebUI界面設(shè)計(jì),它將文生圖領(lǐng)域的擴(kuò)散算法模塊化與圖形化,提升了生成效率與資源利用率,也大幅降低了開(kāi)發(fā)者的使用門(mén)檻。用戶(hù)可以通過(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,可以大幅提升模型的易用性和用戶(hù)基礎(chǔ)。用戶(hù)無(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>

免責(zé)聲明:此文內(nèi)容為第三方自媒體作者發(fā)布的觀察或評(píng)論性文章,所有文字和圖片版權(quán)歸作者所有,且僅代表作者個(gè)人觀點(diǎn),與極客網(wǎng)無(wú)關(guān)。文章僅供讀者參考,并請(qǐng)自行核實(shí)相關(guān)內(nèi)容。投訴郵箱:editor@fromgeek.com。

極客網(wǎng)企業(yè)會(huì)員

免責(zé)聲明:本網(wǎng)站內(nèi)容主要來(lái)自原創(chuàng)、合作伙伴供稿和第三方自媒體作者投稿,凡在本網(wǎng)站出現(xiàn)的信息,均僅供參考。本網(wǎng)站將盡力確保所提供信息的準(zhǔn)確性及可靠性,但不保證有關(guān)資料的準(zhǔn)確性及可靠性,讀者在使用前請(qǐng)進(jìn)一步核實(shí),并對(duì)任何自主決定的行為負(fù)責(zé)。本網(wǎng)站對(duì)有關(guān)資料所引致的錯(cuò)誤、不確或遺漏,概不負(fù)任何法律責(zé)任。任何單位或個(gè)人認(rèn)為本網(wǎng)站中的網(wǎng)頁(yè)或鏈接內(nèi)容可能涉嫌侵犯其知識(shí)產(chǎn)權(quán)或存在不實(shí)內(nèi)容時(shí),應(yīng)及時(shí)向本網(wǎng)站提出書(shū)面權(quán)利通知或不實(shí)情況說(shuō)明,并提供身份證明、權(quán)屬證明及詳細(xì)侵權(quán)或不實(shí)情況證明。本網(wǎng)站在收到上述法律文件后,將會(huì)依法盡快聯(lián)系相關(guān)文章源頭核實(shí),溝通刪除相關(guān)內(nèi)容或斷開(kāi)相關(guān)鏈接。

2024-06-06
騰訊混元文生圖開(kāi)源模型推理再加速,可支持ComfyUI圖形化工作流
騰訊混元文生圖開(kāi)源模型推理再加速,可支持ComfyUI圖形化工作流

長(zhǎng)按掃碼 閱讀全文