跳到正文
原文
Hugging Face Blog·· 2023-03-09精选AI 评分62

trl 与 peft 集成:在 24GB 消费级 GPU 上 RLHF 微调 20B LLM

Fine-tuning 20B LLMs with RLHF on a 24GB consumer GPU

AI 导读

Hugging Face 正式发布 trl 与 peft 的集成,让大语言模型的 RLHF 微调更易上手。文章以 gpt-neox-20b 为例,通过 8-bit 加载、低秩适配器和 disable_adapters 复用同一模型,在 24GB NVIDIA 4090 上完成微调,完整训练则在单张 A100 上运行。作者指出该方案的主要代价是训练速度约慢一倍,后续将探索多 GPU 扩展与提速。

推荐理由

trl 与 peft 集成后,读者可了解在 24GB 消费级 GPU 上完成 20B 模型 RLHF 微调的具体步骤与内存取舍。

来源:Hugging Face Blog · huggingface.co