diffusion-pipe
https://github.com/tdrussell/diffusion-pipe
tdrussell/diffusion-pipe
拡散モデル用のパイプライン並列トレーニング スクリプト。
現在、
SDXL
、
Flux
、
LTX-Video
、
HunyuanVideo
(t2v)、Cosmos、
Lumina-Image 2.0
、
Wan2.1
(t2v および i2v)、
Chroma
をサポートしています。
#Nvidia_Cosmos
Windows未対応(wsl2では使える)
Block swappingに対応して、動画モデルでも家庭用GPU(VRAM24GB)でもトレーニング可能になった
multi gpuとBlockSwapの両方に対応(同時には使えないハズ)しているため付け外し可能なgpuを複数持つ家庭や今後増設予定のある家庭では便利かもしれない
関連
sd-scripts
AI-Toolkit
Musubi Tuner
SimpleTuner