Upload pattern_dict.py with huggingface_hub
Browse files- pattern_dict.py +10 -0
pattern_dict.py
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
PATTERN_DICT = {
|
| 2 |
+
r"double_blocks\.\d+": (
|
| 3 |
+
"img_attn.proj", "img_attn.qkv", "img_mlp.down_proj", "img_mlp.gate_proj", "img_mlp.up_proj", "img_mod.lin",
|
| 4 |
+
"txt_attn.proj", "txt_attn.qkv", "txt_mlp.down_proj", "txt_mlp.gate_proj", "txt_mlp.up_proj", "txt_mod.lin"
|
| 5 |
+
),
|
| 6 |
+
r"single_blocks\.\d+": (
|
| 7 |
+
"linear1", "linear2",
|
| 8 |
+
"modulation.lin"
|
| 9 |
+
),
|
| 10 |
+
}
|