Hugging Face BlogFeb 23, 2024, 12:00 AM重要 80
Fine-Tuning Gemma Models in Hugging Face
Hugging Face 發布 Gemma 模型微調指南,介紹如何使用 PEFT、QLoRA 與 TRL 高效訓練 Google 的開源大模型。
本指南介紹如何在 Hugging Face 生態系中微調 Google 的 Gemma 開源模型(2B 與 7B)。文章詳細說明了如何利用 PEFT(參數高效微調)技術,特別是 QLoRA(4-bit 量化微調),在消費級 GPU 上進行訓練。透過結合 transformers、peft 與 trl(SFTTrainer)等套件,開發者可以輕鬆載入模型、設定 LoRA 參數、格式化數據集,並將微調後的權重上傳至 Hugging Face Hub,是實作 Gemma 微調的必讀教學。
想看英文原文 / 完整內容?
前往 Hugging Face Blog 原文 →摘要由 AI 整理,以原文為準。