FluxGymでlibraryのimportエラーがでる

今回の話とは全く関係のない画像

既存のモデルで出しづらい画像があったのでちょっとLoRAを作ってみようと、Stability MatrixにFluxGymを入れてみました。

インストールそのものは何のこともなく、WebのUIが立ち上がってトレーニングを開始したのですが、train.bat実行後に以下のメッセージが出ました。

ModuleNotFoundError: No module named 'library'

pip install されてないのかなぁ、とか思ったらそうではなくsd-scriptフォルダにlibraryはありました。

sd-scriptディレクトリ

ということは実行されているパスから見えないということなので、flux_train_network.py に以下の追記をして無理やり解決しました。

flux_train_network.pyの一部

本来的にはワーキングディレクトリを指定してやるのがいいのかなぁとか思っているもののとりあえず応急処置ということで。

実行できましたが、GPUが使われていないと以下のメッセージ。

The installed version of bitsandbytes was compiled without GPU support. 8-bit optimizers, 8-bit multiplication, and GPU quantization are unavailable.

調べるとtorchがCPUしか使わない状態だったので、以下のコマンドでインストールを行う

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

※参考URL：https://pytorch.org/ cudaのバージョンは各環境に合わせてください。

ここまでで実行できるようになり、さてトレーニングをしたのですがなかなか望む結果にはなりません。繰り返しを増やせばいいのか、エポックを増やせばいいのか。単に増やすだけでは時間もかかり過学習になるようですし。VRAM増やしたい・・・