ここはPFGの重みファイル置き場です。pfgを直接cloneしてgenerate.pyで使うか、Colabでとりあえず試すことができます。

wd-v1-4-vit-tagger-v2-last-pooling-layer.onnx

最後のプール層を出力するためのonnxファイルです。wd-v1-4-vit-tagger-v2のonnxファイルから作成されています。

wd14-n10

waifu-diffusionを使って学習させたものです。トークン数は10個です。 SD2系のほかのモデル(v_predに関わらず)でもある程度使える可能性があります。学習データは61万枚ですべて1girlを含んでいるので、複数キャラクターには対応できないかもしれません。また版権キャラはあんまり再現できないと思います。元のモデルもそんなに再現できないので。好きなキャラを生成したい人はそれ用に学習してください。

生成例：

※著作権的なあれのため、入力した画像はwaifu-diffusionで作成したものです。そのため通常の画像よりも再現度が高いです。基本的にwaifu-diffusionで生成できない画像はこの手法でも生成できません。

wd15beta2-n10

It is resumed from wd14-n10 for waifu-diffusion-1.5-beta2 on same dataset.

Licence

このファイル単体で動くものではないので、利用する学習済みモデルのライセンスに従ってください。