File size: 2,094 Bytes
9cceebf
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "2a95c50f3b178e8f49404604c1b8426e",
  "model_name": "moss_image_dit",
  "model_class": "diffsynth.models.moss_image_dit.MOSSImageDiT",
  "quant_config": {
    "load_prequantized": true,
    "configs": [
      {
        "method": "entropack_lossy_quant_int8",
        "backend_config_kwargs": {
          "target_bpp": 6.854,
          "row_rdo_iterations": 2,
          "row_rdo_candidates": 20,
          "use_convrot": true,
          "convrot_groupsize": 256
        },
        "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",
          "final_linear_video",
          "video_head_modulation"
        ]
      },
      {
        "method": "entropack_lossy_quant",
        "backend_config_kwargs": {
          "target_bpp": 8.5,
          "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",
          "final_linear_video",
          "video_head_modulation"
        ]
      }
    ]
  }
}