AI繪圖模型訓練到推理,一個網頁就能搞定
不會寫代碼,也能訓練AI繪畫模型了!
只要借助這個框架,從訓練到推理都能一站式搞定,還能一次管理多個模型。
阿里巴巴團隊推出并開源了這款萬能圖片生成工作臺SCEPTER Studio。
有了它,不用代碼,直接在Web界面當中就能完成模型的訓練與微調,并管理相關數據。
團隊還推出了內置三種模型的DEMO,可以在線體驗SCEPTER的推理功能。
那么SCEPTER具體都能干啥,下面就來一睹為快!
一站式管理繪圖模型
有了SCEPTER,不再需要編寫程序,只要在Web頁面中選擇要訓練的模型,調整好相關的參數,就可以完成模型的訓練和微調了。
具體來看,模型方面,SCEPTER目前支持Stable Diffusion系列的SD 1.5/2.1和SDXL模型。
微調方式上,它支持傳統的全量微調、LoRA等方式,以及自家的SCEdit微調框架,未來還會加入對Res-Tuning調優方式的支持。
SCEdit通過跳過連接的方式,讓擴散模型能以更高的效率完成圖片生成任務,相比LoRA節約了30%-50%的內存開銷。
訓練數據方面,SCEPTER內置了包含3D、日漫、油畫、素描等6種風格,每種風格30組圖像-文本對的自創數據集。
同時,也可將自行準備的圖片打包壓縮,并以csv或txt文檔建立圖像(文件名)和prompt的對應關系,導入SCEPTER平臺。
如果連文檔也不想寫,也可以在SCEPTER界面中直接上傳圖片并標注Prompt,管理、添加、刪除數據集中的圖片。
到了推理階段,SCEPTER支持的下游任務包括文生圖和可控圖像合成,未來還將支持圖像編輯,用法和現有的Web版SD相似。
同時,SCEPTER界面中還集成了咒語書(Prompt合集)和一些現成的微調模型。
那么,SCEPTER該如何食用呢?
如果只想玩玩生成的話,官方在HuggingFace和魔搭社區上的DEMO就可以滿足要求,并且后者還是中文界面。
如果要使用數據管理、訓練等功能,就需要自行安裝部署完整版本了,具體步驟可以參照GitHub頁面中的教程。
整個過程中,只有安裝部署的環節需要用到一些簡單的代碼,之后的一切過程就都可以在Web界面里直接操作了。
感興趣的話,就去體驗一下吧!
傳送門:https://github.com/modelscope/scepter