AX1Y2JP commited on
Commit
e239750
·
verified ·
1 Parent(s): 55cafca

Upload pattern_dict.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
+ }