跳到正文
原文
Hugging Face Blog·· 2024-02-23精选AI 评分65

Hugging Face 发布 Gemma 模型 LoRA/QLoRA 微调教程

Fine-Tuning Gemma Models in Hugging Face

AI 导读

Hugging Face 发布 Gemma 模型微调教程,演示用 Transformers、PEFT 与 trl 的 SFTTrainer,通过 LoRA 和 4-bit QLoRA(bitsandbytes、nf4 量化)在 google/gemma-2b 上以 english_quotes 数据集训练模型按指定格式输出名言和作者。

推荐理由

Hugging Face 官方用 LoRA 和 4-bit QLoRA 给出 Gemma 微调的完整可复现代码,覆盖 GPU 和 TPU 两种路径。

来源:Hugging Face Blog · huggingface.co