Update README.md
Browse files
README.md
CHANGED
|
@@ -30,15 +30,13 @@ This is the `pattern_dict` for compression:
|
|
| 30 |
|
| 31 |
```python
|
| 32 |
pattern_dict_comfyui = {
|
| 33 |
-
r"distilled_guidance_layer\.layers\.\d+": (
|
| 34 |
-
"in_layer",
|
| 35 |
-
"out_layer"
|
| 36 |
-
),
|
| 37 |
r"double_blocks\.\d+": (
|
|
|
|
| 38 |
"img_attn.qkv",
|
| 39 |
"img_attn.proj",
|
| 40 |
"img_mlp.0",
|
| 41 |
"img_mlp.2",
|
|
|
|
| 42 |
"txt_attn.qkv",
|
| 43 |
"txt_attn.proj",
|
| 44 |
"txt_mlp.0",
|
|
@@ -47,6 +45,7 @@ pattern_dict_comfyui = {
|
|
| 47 |
r"single_blocks\.\d+": (
|
| 48 |
"linear1",
|
| 49 |
"linear2",
|
|
|
|
| 50 |
),
|
| 51 |
}
|
| 52 |
```
|
|
|
|
| 30 |
|
| 31 |
```python
|
| 32 |
pattern_dict_comfyui = {
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
r"double_blocks\.\d+": (
|
| 34 |
+
"img_mod.lin",
|
| 35 |
"img_attn.qkv",
|
| 36 |
"img_attn.proj",
|
| 37 |
"img_mlp.0",
|
| 38 |
"img_mlp.2",
|
| 39 |
+
"txt_mod.lin",
|
| 40 |
"txt_attn.qkv",
|
| 41 |
"txt_attn.proj",
|
| 42 |
"txt_mlp.0",
|
|
|
|
| 45 |
r"single_blocks\.\d+": (
|
| 46 |
"linear1",
|
| 47 |
"linear2",
|
| 48 |
+
"modulation.lin",
|
| 49 |
),
|
| 50 |
}
|
| 51 |
```
|