原文

PyTorch-Lightningのバージョン2.6.0およびそれ以前には、チェックポイントの読み込み機構において安全でないデシリアライズの脆弱性（CWE-502）が存在します。一般的に保存されたモデルの状態を読み込むために使用されるLightningModule.load_from_checkpoint()メソッドは、内部でtorch.load()を呼び出しますが、安全性を高めるためのweights_only=Trueパラメータを設定していません。このデフォルトの動作により、Pickleモジュールを介して任意のPythonオブジェクトをデシリアライズできます。リモートの攻撃者は、悪意のある細工を施したチェックポイントファイルを提供することでこの脆弱性を悪用し、ファイルが読み込まれた際に被害者のシステム上で任意のコードを実行する可能性があります。