SSD-1B-anime / README.md
furusu's picture
Update README.md
17c6379
|
raw
history blame
1.65 kB
metadata
tags:
  - text-to-image

このモデルは以下の2ステップで作成されました。

  1. SSD-1BNekorayXLsdxl-1.0の差分の1.3倍でマージ。蒸留前と蒸留後のkeyについてはこのマッピングを想定しています。
  2. NekorayXLの最終出力との差を損失にして蒸留(学習率1e-5,バッチサイズ4で9000ステップ)

なんだかよさげな雰囲気だがanatomyに問題を抱えている。

SSD-1BとSDXLのkey対応について

削除したモジュールがどれか分からないので、コサイン類似度を利用して推定しました。 Transformer_depthだけ変わっているので(多分)Attention層のパラメータをSDXLとSSD-1B調査しました。 2層⇒1層となる場合先頭の層が残ります。 10層⇒4層となる場合1,2,3,7番目の層が残ります。

※up層の3番目は10層のままですが、コサイン類似度の結果が不可解なものになっていました。とりあえずここは変更されていないと仮定しています。

image/png image/png