File size: 2,074 Bytes
a02c165
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
{
  "model_hash": "770a2d4aebc6ebd39749c1186a16d8da",
  "model_name": "moss_image_dit",
  "model_class": "diffsynth.models.moss_image_dit.MOSSImageDiT",
  "quant_config": {
    "load_prequantized": true,
    "configs": [
      {
        "method": "entropack_lossy_quant",
        "backend_config_kwargs": {
          "target_bpp": 4.417207792207792,
          "row_rdo_iterations": 2,
          "row_rdo_candidates": 20
        },
        "exclude_modules": [
          "video_embedder",
          "text_embedder.proj",
          "text_embedder.blocks.0.qkv",
          "text_embedder.blocks.0.proj",
          "text_embedder.blocks.0.ffn_gate",
          "text_embedder.blocks.0.ffn_up",
          "text_embedder.blocks.0.ffn_down",
          "text_embedder.blocks.1.qkv",
          "text_embedder.blocks.1.proj",
          "text_embedder.blocks.1.ffn_gate",
          "text_embedder.blocks.1.ffn_up",
          "text_embedder.blocks.1.ffn_down",
          "time_embedding.0",
          "time_embedding.2",
          "shared_block_modulation",
          "video_head_modulation",
          "final_linear_video",
          "proj.o"
        ]
      },
      {
        "method": "entropack_lossy_quant",
        "backend_config_kwargs": {
          "target_bpp": 5.0,
          "row_rdo_iterations": 2,
          "row_rdo_candidates": 20
        },
        "target_modules": [
          "video_embedder",
          "text_embedder.proj",
          "text_embedder.blocks.0.qkv",
          "text_embedder.blocks.0.proj",
          "text_embedder.blocks.0.ffn_gate",
          "text_embedder.blocks.0.ffn_up",
          "text_embedder.blocks.0.ffn_down",
          "text_embedder.blocks.1.qkv",
          "text_embedder.blocks.1.proj",
          "text_embedder.blocks.1.ffn_gate",
          "text_embedder.blocks.1.ffn_up",
          "text_embedder.blocks.1.ffn_down",
          "time_embedding.0",
          "time_embedding.2",
          "shared_block_modulation",
          "video_head_modulation",
          "final_linear_video",
          "proj.o"
        ]
      }
    ]
  }
}