存模式:4GB顯存舊本穩(wěn)定跑SD圖像修復(fù))
IOPaint低內(nèi)存模式4GB顯存舊本穩(wěn)定跑SD圖像修復(fù)【免費(fèi)下載鏈接】IOPaintImage inpainting tool powered by SOTA AI Model. Remove any unwanted object, defect, people from your pictures or erase and replace(powered by stable diffusion) any thing on your pictures.項(xiàng)目地址: https://gitcode.com/GitHub_Trending/io/IOPaint打開IOPaint跑Stable Diffusion重繪任務(wù)顯存幾秒鐘飆到90%生成階段直接OOM。在4GB顯存的舊筆記本上IOPaint的低內(nèi)存模式Low Memory Modelow_mem參數(shù)就是為這種場景準(zhǔn)備的。核心概念速覽開啟低內(nèi)存模式后模型管線會(huì)執(zhí)行兩項(xiàng)操作注意力切片和VAE分塊tiling。前者把注意力計(jì)算拆成小塊分段執(zhí)行后者讓VAE變分自編碼器負(fù)責(zé)把潛變量解碼回像素圖的組件按塊解碼圖像而不是整圖一次性解碼顯存峰值隨之下降。相當(dāng)于把整圖一次性加載的搬運(yùn)方式改成了分批配送。實(shí)現(xiàn)入口在 iopaint/model/utils.py 的enable_low_mem函數(shù)中各擴(kuò)散模型SD、SDXL、ControlNet等加載時(shí)統(tǒng)一調(diào)用。上手指南在設(shè)置界面勾選低內(nèi)存模式推薦啟動(dòng)IOPaint打開設(shè)置頁面切換到模型Model選項(xiàng)卡。勾選 Low Memory 復(fù)選框界面定義見 iopaint/web_config.py 中的low_mem控件。若顯存仍然吃緊可同時(shí)勾選 CPU Offload模型權(quán)重放到內(nèi)存與 CPU Text Encoder文本編碼器放CPU。保存配置后重啟再執(zhí)行修復(fù)任務(wù)。命令行啟動(dòng)IOPaint低內(nèi)存模式命令行或批量處理場景下直接在啟動(dòng)命令中追加參數(shù)即可選項(xiàng)定義見 iopaint/cli.pypython -m iopaint start --low_mem --model runwayml/stable-diffusion-inpainting顯存不夠時(shí)再加--cpu_offload --cpu_textencoder三個(gè)選項(xiàng)互相獨(dú)立可以按需組合。效果驗(yàn)證指標(biāo)標(biāo)準(zhǔn)模式低內(nèi)存模式low_memTrue顯存占用偏高低顯存卡易OOM降低切片分塊解碼設(shè)備覆蓋自動(dòng)化回歸CUDA / Apple SiliconCUDA / Apple Silicon / 純CPU對比結(jié)論來自 iopaint/tests/test_low_mem.py 的自動(dòng)化回歸low_memTrue時(shí)同一個(gè) runwayml/stable-diffusion-inpainting 模型在 CUDA、Apple SiliconMPS和純CPULCM采樣5步三種設(shè)備上均通過生成校驗(yàn)。避坑指南為什么生成的圖全黑或發(fā)綠多半是FP16半精度計(jì)算的兼容性問題。勾選--no_half切回全精度FP32占的顯存略多出圖顏色恢復(fù)正常。開啟后生成為什么變慢了切片和分塊會(huì)帶來額外的搬運(yùn)開銷速度比標(biāo)準(zhǔn)模式略慢換的是更低的顯存峰值。顯存夠用時(shí)關(guān)掉它即可。開啟后仍然OOM怎么辦再疊加--cpu_offload把模型權(quán)重移到內(nèi)存或--cpu_textencoder讓文本編碼器跑在CPU上兩個(gè)選項(xiàng)在設(shè)置界面都有對應(yīng)勾選框。純CPU機(jī)器能跑嗎可以。測試文件已驗(yàn)證 low_mem 開啟時(shí) SD 1.5 配合LCM采樣能在CPU上跑通但速度有限只適合處理小圖。適用邊界適合誰4GB顯存左右的舊筆記本或亮機(jī)卡邊跑模型邊掛著瀏覽器等多任務(wù)的并行場景批量處理512×512及以下的小圖只有CPU、沒有獨(dú)顯的機(jī)器。不適合誰1024×1024以上的大圖生成切片后速度損失明顯要求實(shí)時(shí)響應(yīng)的交互創(chuàng)作流程多個(gè)大模型串聯(lián)的復(fù)雜pipeline顯存本來就夠、追求速度的場景。結(jié)尾顯存告急時(shí)把低內(nèi)存模式打開是第一件該做的事。更多參數(shù)說明見 README.md?!久赓M(fèi)下載鏈接】IOPaintImage inpainting tool powered by SOTA AI Model. Remove any unwanted object, defect, people from your pictures or erase and replace(powered by stable diffusion) any thing on your pictures.項(xiàng)目地址: https://gitcode.com/GitHub_Trending/io/IOPaint創(chuàng)作聲明:本文部分內(nèi)容由AI輔助生成(AIGC),僅供參考