LLMのファインチューニングのためのツール「Axolotl」の概要をまとめました。 1. Axolotl「Axolotl」は、LLMのファインチューニングのためのツールです。様々なLLM、データセット形式、アーキテクチャをサポートします。 ・「LLaMA」「Pythia」「Falcon」「MPT」などの様々なHuggingFaceモデルを学習 ・「Full Finetune」「Lora」「Kalora」Relora」「Guptak」をサポート ・単純なyamlやCLIオーバーライドで構成をカスタマイズ ・様々なデータセット形式をロードしたり、カスタム形式を使用したり、独自のトークン化されたデータセットを使用可能 ・「xFormer」「Flash attention」「Rope scaling」「MultiPacking」と統合 ・「FSDP」または「DeepSpeed」を介して単一GPUま
![LLMのファインチューニングのためのツール Axolotl|npaka](https://cdn-ak-scissors.b.st-hatena.com/image/square/ec8133304784946a92a03c1675c147c6d9cff8c5/height=288;version=1;width=512/https%3A%2F%2Fassets.st-note.com%2Fproduction%2Fuploads%2Fimages%2F120373686%2Frectangle_large_type_2_9c351bc8d1f401965008f70d021688e7.png%3Ffit%3Dbounds%26quality%3D85%26width%3D1280)