Soup MakazhanAlpamys
Fine-tune LLMs from one YAML. Layer streaming trains an 8B model on a 4 GB laptop GPU.
- 主要言語
- Python
- Stars
- 1,684
- Forks
- 262
タグ
- CLIツール
- ディープラーニング
- ローカルAI
- 省メモリ
- AIワークフロー
概要
YAML設定1枚と1コマンドでLLMのファインチューニングを実現するCLIツール。レイヤーストリーミング技術で凍結したベースモデルをVRAM外に置き、デコーダ層ごとにGPUへ供給することで、8Bクラスのモデルを4GB VRAMのノートPC GPUでも学習できる。QLoRA/NF4量子化やDPO/ORPO/SimPO/KTOなどの選好学習、評価ゲート(soup ship)も内蔵し、SSHやクラウド不要のローカル学習に特化している。
README
<p align="center"> <img src="soup.png" alt="Soup" width="280"> </p> <h1 align="center">Soup</h1> <p align="center"> <strong>LLMのファインチューニングとポストトレーニングを1コマンドで。SSH不要、設定地獄も不要。</strong> </p> <p align="center"> <a href="https://trysoup.dev">Webサイト</a> · <a href="#quick-start">クイックスタート</a> · …