File size: 789 Bytes
40ef13a 27c6a99 40ef13a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
{
"dfloat11_config": {
"version": "0.5.0",
"threads_per_block": [
512
],
"bytes_per_thread": 8,
"pattern_dict": {
"distilled_guidance_layer\\.layers\\.\\d+": [
"linear_1",
"linear_2"
],
"transformer_blocks\\.\\d+": [
"attn.to_q",
"attn.to_k",
"attn.to_v",
"attn.add_k_proj",
"attn.add_v_proj",
"attn.add_q_proj",
"attn.to_out.0",
"attn.to_add_out",
"ff.net.0.proj",
"ff.net.2",
"ff_context.net.0.proj",
"ff_context.net.2"
],
"single_transformer_blocks\\.\\d+": [
"proj_mlp",
"proj_out",
"attn.to_q",
"attn.to_k",
"attn.to_v"
]
}
}
} |