sd-scrips 実行時に「RuntimeError: NaN detected in latents: (画像ファイル)」 のエラーが発生する

Hiromi

質問: sd-scriptsでの画像のエラー

sd-scriptsを実行したら「RuntimeError: NaN detected in latents: (画像ファイル名のパス)」エラーになってしまい、 実行できませんでした。対処法はありますか?

エラーについて

sd-scripts実行時に次のエラーが発生します。

原因

いくつかの原因がありますが、学習元の画像ファイルが破損していない場合には、VAEが原因の可能性があります。
画像のlatentを求める際に、pretrained_model_name_or_path で指定しているcheckpointのVAEが利用されることがあり、 その影響で、latent計算の際に数値が壊れて NaNの値が発生します。

対処法

ほかのモデル(checkpoint)を利用する

学習のベースモデルをほかのモデルに変更します。
pretrained_model_name_or_path で指定しているcheckpointを別のモデルのcheckpointに変更します。

著者

とりっち

iPentecのメインデザイナー
Webページ、Webクリエイティブのデザインを担当。PhotoshopやIllustratorの作業もする。 最近は生成AIの画像生成の沼に沈んでいる。