text-generation-inference 文件

模型安全。

Hugging Face's logo
加入 Hugging Face 社群

並獲得增強的文件體驗

開始使用

模型安全。

Pytorch 預設使用 pickle,這意味著在很長一段時間內,*所有*使用該格式的模型在純粹載入模型時都可能執行意外程式碼。

Python 的 pickle 頁面上有一個很大的紅色警告連結,但社群在很長一段時間內都忽略了它。現在 AI/ML 得到了更廣泛的使用,我們需要擺脫這種格式。

HuggingFace 在這方面發揮了主導作用,它建立了一種包含純資料的新格式(safetensors),並正在緩慢但穩步地將所有庫預設使用它。此舉是故意的緩慢,以便儘可能減少對使用者造成破壞性更改的影響。

TGI 2.0

自 TGI 2.0 釋出以來,我們藉此主要版本升級的機會,打破了對這些 pytorch 模型的向後相容性(因為它們對部署它們的人來說存在巨大的安全風險)。

從現在起,TGI 將不會自動轉換 pickle 檔案,除非在環境變數中設定了 --trust-remote-code 標誌或 TRUST_REMOTE_CODE=true。此標誌已用於社群定義的推理程式碼,因此它很好地代表了您對模型提供商的信任程度。

如果您想使用使用 pickle 的模型,但仍不想完全信任作者,我們建議您使用我們為此目的製作的空間進行轉換。

https://huggingface.co/spaces/safetensors/convert

這個空間將對原始模型建立一個 PR,無論原始作者的合併狀態如何,您都可以直接使用它。只需使用

docker run .... --revision refs/pr/#ID # Or use REVISION=refs/pr/#ID in the environment
< > 在 GitHub 上更新

© . This site is unofficial and not affiliated with Hugging Face, Inc.