1枚に載らないモデルを動かす
第3章では、文章を1トークンずつ作る速さが、重みをメモリから読む速さでほぼ決まることを見ました。この章ではその手前の問題、つまり重みがそもそもGPUのメモリに入らないときにどうするかを考えます。図のように、700億パラメータ(70B)のモデルを16ビットで持つと約140GB。青い箱で描いたGPU 1枚のメモリ(例として80GB)から、紫のバーが大きくはみ出します。
打つ手は大きく2つです。1つは、重み1つあたりのビット数を減らして全体を小さくする量子化。もう1つは、何枚ものGPUに分けて持つ並列化です。まず、メモリに何がどれだけ載るのかを足し算するところから始めます。



