LoRA アダプターに Serverless Inference を使用する理由
LoRA アダプター向けの Serverless Inference には、次の利点があります。- 一度アップロードすれば、サーバーを管理せずにデプロイできます。
- アーティファクトのバージョン管理により、稼働中のバージョンをトラッキングできます。
- モデルの重み全体ではなく、小さな LoRA アダプターのファイルを差し替えることでモデルを更新できます。
ワークフロー
大まかには、カスタム LoRA アダプターをサーブするには 3 つのステップがあります:- LoRA アダプターの重みを Weights & Biases アーティファクトとしてアップロードします。
- アーティファクトの URI を API のモデル名として指定します。
- CoreWeave は推論のために重みを動的にロードします。
前提条件
以下が必要です。- CoreWeave Forge APIキー。
- Weights & Biases project。
openaiおよびwandbパッケージがインストールされた Python 3.8 以降:pip install wandb openai。
LoRA アダプターを追加して使用する
以下の方法で、CoreWeave Forge アカウントに LoRA アダプターを追加して使い始めることができます。LoRA アダプターをトレーニングした場所に応じて、該当するタブを選択してください。- 他の場所でトレーニングした LoRA アダプターをアップロードする
- Serverless RL で新しい LoRA アダプターをトレーニングする
- 教師ありファインチューニングで新しい LoRA アダプターをトレーニングする
独自のカスタム LoRA アダプターのディレクトリを、Weights & Biases のアーティファクトとしてアップロードします。LoRA アダプターを他の場所 (ローカル環境、クラウドプロバイダー、パートナーサービスなど) でトレーニングした場合は、この方法を使用します。次の Python コードは、ローカルに保存されている LoRA アダプターの重みを、バージョン管理されたアーティファクトとして Weights & Biases にアップロードします。必須のメタデータ (ベースモデルとストレージリージョン) を含む
lora タイプのアーティファクトを作成し、ローカルディレクトリから LoRA アダプターのファイルを追加したうえで、推論で使用できるように Weights & Biases の project にログします。既存の LoRA アダプターを使用するための主な要件
独自の LoRA アダプターを Serverless Inference で使用するには、次の要件を満たしていることを確認してください。- LoRA アダプターが、サポートされるベースモデルセクションに記載されているいずれかのモデルでトレーニングされていること。
- LoRA アダプターの rank が、サポートされるベースモデルの表に記載されている、対応するベースモデルの 最大 LoRA アダプター rank 以下であること。
- LoRA アダプターが PEFT 形式で保存され、CoreWeave Forge アカウント内に
loraタイプのアーティファクトとして存在すること。 - 低レイテンシーを実現するため、LoRA アダプターが
storage_region="coreweave-us"に保存されていること。 - アップロード時に、トレーニングに使用したベースモデルの名前 (例:
meta-llama/Llama-3.1-8B-Instruct) を含めること。これにより、Serverless Inference が正しいモデルで LoRA アダプターを読み込めるようになります。
サポートされるベースモデル
LoRA アダプターは、次のいずれかのベースモデルに対してトレーニングされている必要があります。推論時に Serverless Inference がアダプターを正しいベースモデルと組み合わせられるように、wandb.base_model を設定する際は正確なモデル ID の文字列を使用してください。
最大 LoRA アダプター rank 列は、Serverless Inference が各ベースモデルで読み込めるアダプターの最大サイズを示します。この値は、他の環境でトレーニングしてアップロードするアダプターに適用される上限であり、トレーニングの設定項目ではありません。Serverless SFT および Serverless RL で作成されたアダプターは固定の rank を使用し、この上限を超えることはありません。詳細は LoRA rank と alpha を参照してください。