mingyi456 commited on
Commit
13146ea
·
1 Parent(s): 40518fd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -15
README.md CHANGED
@@ -31,19 +31,23 @@ This is the `pattern_dict` for compressing Chroma-based models in ComfyUI:
31
 
32
  ```python
33
  pattern_dict_comfyui = {
34
- "double_blocks\.\d+": (
35
- "img_attn.qkv",
36
- "img_attn.proj",
37
- "img_mlp.0",
38
- "img_mlp.2",
39
- "txt_attn.qkv",
40
- "txt_attn.proj",
41
- "txt_mlp.0",
42
- "txt_mlp.2",
43
- ),
44
- "single_blocks\.\d+": (
45
- "linear1",
46
- "linear2",
47
- ),
48
- }
 
 
 
 
49
  ```
 
31
 
32
  ```python
33
  pattern_dict_comfyui = {
34
+ r"distilled_guidance_layer\.layers\.\d+": (
35
+ "in_layer",
36
+ "out_layer"
37
+ ),
38
+ r"double_blocks\.\d+": (
39
+ "img_attn.qkv",
40
+ "img_attn.proj",
41
+ "img_mlp.0",
42
+ "img_mlp.2",
43
+ "txt_attn.qkv",
44
+ "txt_attn.proj",
45
+ "txt_mlp.0",
46
+ "txt_mlp.2",
47
+ ),
48
+ r"single_blocks\.\d+": (
49
+ "linear1",
50
+ "linear2",
51
+ ),
52
+ }
53
  ```