快轉到主要內容

Text Generation WebUI:啟動引數

開源AI人工智慧應用 大型語言模型
✍ 切換正體/簡體字
Ivon Huang
作者
Ivon Huang
Text Generation WebUI使用教學 - 系列文章
系列之 5: 本篇文章

Text Generation WebUI因為整合了許多技術,有時需要設定啟動引數(arguments)才能使用特定硬體計算。

預設模式下,Text Generation WebUI會使用GPU計算,您可以設定引數強迫使用CPU計算。語言模型即使全用CPU跑,產生對話的速度還是可以接受的。

  1. 於終端機按Ctrl+C,終止Text Generation WebUI程式。

  2. text-generation-webui-main資料夾,編輯CMD_FLAGS.txt

  3. 在最後一行填入引數,比如:

--listen --auto-devices
  1. 完整引數請參考開發者的 Github,下面我介紹一些常用的。
  • --auto-devices引數自動分配計算任務給CPU和GPU
  • --cpu:強制只用CPU計算
  • --gpu-memory設定最多分配的VRAM,防止記憶體不足之用,數值視您的顯卡VRAM而定
  • --model指定要載入的模型
  • --listen使WebUI的port可從外部網路存取
  1. 範例,我將引數設定如下:
--listen --auto-devices --gpu-memory 2 --model "模型檔名"
  1. 儲存CMD_FLAGS.txt,重新啟動WebUI。啟動後會自動判別模型種類,選取適合的方法載入。

Text Generation WebUI使用教學 - 系列文章
系列之 5: 本篇文章

相關文章

Text Generation WebUI安裝教學(Linux/Windows/macOS)
開源AI人工智慧應用 大型語言模型 Self-hosting自架 LLaMA AI Large Language Model
Text Generation WebUI:下載大型語言模型(LLM)
開源AI人工智慧應用 大型語言模型 LLaMA AI Large Language Model
Text Generation WebUI硬體需求
開源AI人工智慧應用 大型語言模型

留言板

此處提供二種留言板。點選按鈕,選擇您覺得方便的留言板。要討論程式碼請用Giscus,匿名討論請用Disqus。

這是Giscus留言板,需要登入Github帳號才能留言。支援markdown語法,方便您張貼程式碼,若要上傳圖片請貼Imgur連結。您的留言會在Github Discussions向所有人公開。

這是Disqus留言板,您可能會看到Disqus強制投放的廣告。無論有無登入皆可留言。此處留言只有本站參與者看得到。若您選擇以訪客身份匿名留言,就不會收到後續回覆通知。