軽量版のDreambooth on Stable Diffusionを作った人が。ざっくり見たところtrainerのacceleratorをddpから無指定(=auto、つまりgpu?)にして、CrossAttentionのeinsumを分割して行うようにしたみたい。後者で遅くなる代わりにメモリ使用量が減るのか。精度は変わらないはず。https://github.com/gammagec/Dreambooth-SD-optimized …
People in Hollywood are so vain.
So self-centered.
Most offices I walk into just have posters of their own movies on the walls!
Instead, I'm gonna print these classic moments out -- to inspire my next films.
A thread of just images (1/5)
さっきの軽量版と同じeinsumの分割対応を手元で入れたらVRAM使用量23.5GB→22.5GBに微減。速度は1.5倍くらいに遅くなった。ただバッチサイズ2でも使用量23.6GBで回るようになった(・∀・) トータルの速度は元のバージョンと同じくらいだけど、バッチサイズを増やしたぶん精度が上がるかも。
Fine tuning #stablediffusion to make Pokemon!
I wrote a quick guide on fine tuning your own Stable Diffusion: https://github.com/LambdaLabsML/examples/tree/main/stable-diffusion-finetuning …
I also released my Pokemon model, you can try it out on Replicate: https://replicate.com/lambdal/text-to-pokemon …
or with this Notebook: https://github.com/LambdaLabsML/lambda-diffusers/blob/main/notebooks/pokemon_demo.ipynb …
今日のStable Diffusion の改造方法報告スペースのスライドはこのツイートにスレッド状で貼り付けています。聞き逃した方はスライドを見れば何となくわかるかもしれません。 https://twitter.com/alfredplpl/status/1573999912613662723 …
うおっ!?Colabで追加学習が始まった!?と思いきやすぐプロセスキルされた。OOMエラーじゃ無いって事はVRAMじゃなくてメモリが足りなかったんかな?ランタイムタイプはハイメモリなのに…GPUはP100
お~ついにちゃんと学習開始する所が見れた。LambdaLabsで奇跡的にA6000x1インスタンスが空いたから課金しちゃったよ。これ動かしても例のポケモンAIが出来るだけなんだが…まあ勉強代として許容します。僕が寝てる間に頑張って学習してくれ
LambdaLabs上で学習したチェックポイントを回収しようとしたらダウンロード速度が300k~1Mbpsしか出なくて、チェックポイントは14GBあるから全く終わらねえ。エグい罠やんけ
おお!RunpodのA5000で学習できた!ひとまず0エポック目を突破。つまりVRAMが24GBでも行けたという事で、それならRTX3090でも行けるという事になる。するとどうなるかというと、ご家庭のPCでもRTX3090や4090があればSDを学習させられる。クラウドでもA6000は全然空いてないけどA5000や3090は借りやすい
AIがキャラをちゃんと描けるの凄すぎてマジクソビビる。ただし、こんな芸当ができるのは一部のキャラだけっぽい。美樹さやかさんの事はtrinart君は相当うろ覚えだった。まどかちゃんはこんな風にちゃんと描けるクセに。AIオイ!知識の偏り! →RT
Dreambooth on SDの今の個人的まとめ
・キャラを学ばせる場合、学習データからタッチが異なるものを除き統一するとスタイルが強くなる(スタイルが弱いと実写寄りの画像が生成されやすい)ボケた画像も避ける
・学習データはわりと多くてもOK。今のところ30~100枚(regularization300~400枚)で学習中
よ~し、RTX4090買って家でAI学習しちゃうぞ~!って意気込んでる人達に朗報です。4090積んだゲーミングPCをぶん回した時の消費電力を800Wとすると、電気代は1時間あたり21円かかります。つまり、一ヶ月ぶっ続けでぶん回すと1万5千円!これはひどい