kelseye commited on
Commit
9cceebf
·
verified ·
1 Parent(s): a02c165

Upload model using ModelScope SDK

Browse files
.gitattributes CHANGED
@@ -44,10 +44,10 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
44
  *.tar filter=lfs diff=lfs merge=lfs -text
45
  *.wasm filter=lfs diff=lfs merge=lfs -text
46
  *.zst filter=lfs diff=lfs merge=lfs -text
47
- *tfevents* filter=lfs diff=lfs merge=lfs -textassets/MOSS-Image-Base/dit_4.5bit.jpg filter=lfs diff=lfs merge=lfs -text
48
- assets/MOSS-Image-Base/dit_int8_6bit.jpg filter=lfs diff=lfs merge=lfs -text
49
- assets/MOSS-Image-SFT/bf16.jpg filter=lfs diff=lfs merge=lfs -text
50
- assets/MOSS-Image-SFT/dit_3.5bit.jpg filter=lfs diff=lfs merge=lfs -text
51
- assets/MOSS-Image-SFT/dit_4.5bit.jpg filter=lfs diff=lfs merge=lfs -text
52
- assets/MOSS-Image-SFT/dit_6bit.jpg filter=lfs diff=lfs merge=lfs -text
53
  assets/MOSS-Image-SFT/dit_int8_6bit.jpg filter=lfs diff=lfs merge=lfs -text
 
 
44
  *.tar filter=lfs diff=lfs merge=lfs -text
45
  *.wasm filter=lfs diff=lfs merge=lfs -text
46
  *.zst filter=lfs diff=lfs merge=lfs -text
47
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
48
+ "/assets/MOSS-Image-Base/dit_int8_6bit.jpg" filter=lfs diff=lfs merge=lfs -text
49
+ "/assets/MOSS-Image-Base/dit_int8_7bit.jpg" filter=lfs diff=lfs merge=lfs -text
50
+ "/assets/MOSS-Image-SFT/dit_int8_6bit.jpg" filter=lfs diff=lfs merge=lfs -text
51
+ "/assets/MOSS-Image-SFT/dit_int8_7bit.jpg" filter=lfs diff=lfs merge=lfs -textassets/MOSS-Image-Base/dit_int8_6bit.jpg filter=lfs diff=lfs merge=lfs -text
 
52
  assets/MOSS-Image-SFT/dit_int8_6bit.jpg filter=lfs diff=lfs merge=lfs -text
53
+ assets/MOSS-Image-SFT/dit_int8_7bit.jpg filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -4,41 +4,49 @@ We provide pre-quantized weight packages produced by [**EntroPack**](https://git
4
 
5
  The packages run with as little as 4 GB of VRAM.
6
 
7
- Base is provided at three levels, SFT additionally ships a 3.5-bit variant, and the 3.5-bit and 4.5-bit DiTs use mixed quantization. Base and SFT share the same text encoder packages:
8
 
9
- * **`3.5bit`** (SFT only, lowest VRAM) — mixed-bit: the quantization-sensitive layers keep **5.0 bpp**, while the remaining layers use **3.2516 bpp** (weighted target 3.5 bpp);
10
- * **`4.5bit`** (balanced) — mixed-bit: the quantization-sensitive layers keep **5.0 bpp**, while the remaining layers use **4.4172 bpp** (weighted target 4.5 bpp);
11
- * **`6bit`** (highest precision) — uniform entropy coding at 6 bits per weight;
12
- * **`int8_6bit`** (fastest inference) — weights are first **INT8 W8A8-quantized**, then entropy-coded by EntroPack down to 6-bit storage; the forward pass uses **INT8 GEMM**, which delivers a faster inference speed than the BF16 GEMM used by the other levels.
 
 
 
 
 
 
 
 
13
 
14
  All packages are entropy-coded from the original BF16 weights directly. Every `*.safetensors` ships with a same-named `*.json` that records its quantization config. `actual bpp = 8 × Σ storage bytes / Σ numel` is measured on the package artifacts.
15
 
16
  ## Preview
17
 
18
- All previews are generated with the same prompt and seed; `bf16` is the full-precision reference.
19
 
20
  ### SFT
21
 
22
  <table width="100%">
23
  <tr>
24
  <td width="33%" align="center">bf16 (reference)</td>
25
- <td width="33%" align="center">6bit</td>
26
  <td width="33%" align="center">int8@6</td>
27
  </tr>
28
  <tr>
29
  <td align="center"><img width="100%" src="assets/MOSS-Image-SFT/bf16.jpg"></td>
30
- <td align="center"><img width="100%" src="assets/MOSS-Image-SFT/dit_6bit.jpg"></td>
31
  <td align="center"><img width="100%" src="assets/MOSS-Image-SFT/dit_int8_6bit.jpg"></td>
32
  </tr>
33
  <tr>
 
34
  <td width="33%" align="center">4.5bit</td>
35
  <td width="33%" align="center">3.5bit</td>
36
- <td></td>
37
  </tr>
38
  <tr>
 
39
  <td align="center"><img width="100%" src="assets/MOSS-Image-SFT/dit_4.5bit.jpg"></td>
40
  <td align="center"><img width="100%" src="assets/MOSS-Image-SFT/dit_3.5bit.jpg"></td>
41
- <td></td>
42
  </tr>
43
  </table>
44
 
@@ -46,16 +54,24 @@ All previews are generated with the same prompt and seed; `bf16` is the full-pre
46
 
47
  <table width="100%">
48
  <tr>
49
- <td width="25%" align="center">bf16 (reference)</td>
50
- <td width="25%" align="center">6bit</td>
51
- <td width="25%" align="center">4.5bit</td>
52
- <td width="25%" align="center">int8@6</td>
53
  </tr>
54
  <tr>
55
  <td align="center"><img width="100%" src="assets/MOSS-Image-Base/bf16.jpg"></td>
56
  <td align="center"><img width="100%" src="assets/MOSS-Image-Base/dit_6bit.jpg"></td>
57
- <td align="center"><img width="100%" src="assets/MOSS-Image-Base/dit_4.5bit.jpg"></td>
 
 
 
 
 
 
 
58
  <td align="center"><img width="100%" src="assets/MOSS-Image-Base/dit_int8_6bit.jpg"></td>
 
 
59
  </tr>
60
  </table>
61
 
@@ -68,7 +84,8 @@ All previews are generated with the same prompt and seed; `bf16` is the full-pre
68
  | dit_3.5bit (mixed) | `models/MOSS-Image-SFT/dit_3.5bit.safetensors` | 3.526 | 2.24 GB |
69
  | dit_4.5bit (mixed) | `models/MOSS-Image-SFT/dit_4.5bit.safetensors` | 4.545 | 2.89 GB |
70
  | dit_6bit | `models/MOSS-Image-SFT/dit_6bit.safetensors` | 6.055 | 3.85 GB |
71
- | dit_int8_6bit (INT8 GEMM) | `models/MOSS-Image-SFT/dit_int8_6bit.safetensors` | 6.063 | 3.85 GB |
 
72
 
73
  ### Base
74
 
@@ -76,7 +93,8 @@ All previews are generated with the same prompt and seed; `bf16` is the full-pre
76
  |---|---|---|---|
77
  | dit_4.5bit (mixed) | `models/MOSS-Image-Base/dit_4.5bit.safetensors` | 4.545 | 2.89 GB |
78
  | dit_6bit | `models/MOSS-Image-Base/dit_6bit.safetensors` | 6.055 | 3.85 GB |
79
- | dit_int8_6bit (INT8 GEMM) | `models/MOSS-Image-Base/dit_int8_6bit.safetensors` | 6.063 | 3.85 GB |
 
80
 
81
  ## Text Encoder
82
 
@@ -86,7 +104,8 @@ The text encoder packages are shared by Base and SFT.
86
  |---|---|---|---|
87
  | text_encoder_4.5bit | `models/text_encoder_4.5bit.safetensors` | 4.552 | 5.97 GB |
88
  | text_encoder_6bit | `models/text_encoder_6bit.safetensors` | 6.070 | 7.29 GB |
89
- | text_encoder_int8_6bit (INT8 GEMM) | `models/text_encoder_int8_6bit.safetensors` | 6.075 | 7.29 GB |
 
90
 
91
  ## Usage
92
 
@@ -104,10 +123,10 @@ cd entropack && pip install -e '.[cuda13]'
104
 
105
  ### Example scripts
106
 
107
- * Inference: [standard](https://github.com/modelscope/DiffSynth-Studio/blob/main/examples/moss_image/model_inference/MOSS-Image-EntroPack.py) / [low VRAM](https://github.com/modelscope/DiffSynth-Studio/blob/main/examples/moss_image/model_inference_low_vram/MOSS-Image-EntroPack.py) — defaults to SFT 4.5bit; switch `dit_package` / `base_model_id` together to use Base or another level
108
  * LoRA training: [train](https://github.com/modelscope/DiffSynth-Studio/blob/main/examples/moss_image/model_training/lora/MOSS-Image-entropack.sh) / [validation](https://github.com/modelscope/DiffSynth-Studio/blob/main/examples/moss_image/model_training/validate_lora/MOSS-Image-entropack.py) — defaults to SFT; replace `MOSS-Image-SFT` with `MOSS-Image-Base` in the dataset and model paths to train on Base
109
 
110
- `dit_package` and `text_encoder_package` are chosen independently: any combination works (e.g. `dit_int8_6bit` with `text_encoder_4.5bit`). The 6-bit packages are pre-registered in DiffSynth-Studio, so they load with a plain `ModelConfig`; the helper below registers any package's json sidecar before loading:
111
 
112
  ```python
113
  import json
@@ -147,9 +166,9 @@ pipe = MOSSImagePipeline.from_pretrained(
147
  )
148
 
149
  image = pipe(
150
- prompt="a cat sitting on a windowsill",
151
  height=1024, width=1024,
152
- seed=42, num_inference_steps=50, cfg_scale=3.0,
153
  )
154
  image.save("image.jpg")
155
  ```
 
4
 
5
  The packages run with as little as 4 GB of VRAM.
6
 
7
+ Base is provided at four levels, SFT additionally ships a 3.5-bit variant. Base and SFT share the same text encoder packages.
8
 
9
+ **BF16 GEMM levels** (the 3.5-bit and 4.5-bit DiTs use mixed quantization):
10
+
11
+ * **`3.5bit`** (SFT only, lowest VRAM) — mixed-bit: the quantization-sensitive layers keep **5.0 bpp**, while the remaining layers use **3.2516 bpp**, giving **3.526 bpp** on the package;
12
+ * **`4.5bit`** (balanced) — mixed-bit: the quantization-sensitive layers keep **5.0 bpp**, while the remaining layers use **4.4172 bpp**, giving **4.545 bpp** on the package;
13
+ * **`6bit`** (highest precision of the BF16 GEMM levels) — uniform entropy coding, **6.055 bpp** on the package.
14
+
15
+ **INT8 GEMM levels** use a hybrid allocation:
16
+
17
+ * **`int8_6bit`** (fastest inference) — the quantization-sensitive linears including embeddings, time embeddings, modulation and final layers are entropy-coded at **8.5 bits per weight** and run on **BF16 GEMM**, while all remaining linears are **INT8 W8A8-quantized with a ConvRot rotation** and run on **INT8 GEMM**; the package averages **5.999 bpp**;
18
+ * **`int8_7bit`** (highest precision of the INT8 levels) — the same hybrid allocation, averaging **7.004 bpp** on the package.
19
+
20
+ The INT8 GEMM path requires Triton and a CUDA GPU with compute capability 8.0 or newer.
21
 
22
  All packages are entropy-coded from the original BF16 weights directly. Every `*.safetensors` ships with a same-named `*.json` that records its quantization config. `actual bpp = 8 × Σ storage bytes / Σ numel` is measured on the package artifacts.
23
 
24
  ## Preview
25
 
26
+ All previews are generated with the same prompt and seed; `bf16` is the full-precision reference. The quantized previews are ordered by PSNR against the `bf16` reference (highest first).
27
 
28
  ### SFT
29
 
30
  <table width="100%">
31
  <tr>
32
  <td width="33%" align="center">bf16 (reference)</td>
33
+ <td width="33%" align="center">int8@7</td>
34
  <td width="33%" align="center">int8@6</td>
35
  </tr>
36
  <tr>
37
  <td align="center"><img width="100%" src="assets/MOSS-Image-SFT/bf16.jpg"></td>
38
+ <td align="center"><img width="100%" src="assets/MOSS-Image-SFT/dit_int8_7bit.jpg"></td>
39
  <td align="center"><img width="100%" src="assets/MOSS-Image-SFT/dit_int8_6bit.jpg"></td>
40
  </tr>
41
  <tr>
42
+ <td width="33%" align="center">6bit</td>
43
  <td width="33%" align="center">4.5bit</td>
44
  <td width="33%" align="center">3.5bit</td>
 
45
  </tr>
46
  <tr>
47
+ <td align="center"><img width="100%" src="assets/MOSS-Image-SFT/dit_6bit.jpg"></td>
48
  <td align="center"><img width="100%" src="assets/MOSS-Image-SFT/dit_4.5bit.jpg"></td>
49
  <td align="center"><img width="100%" src="assets/MOSS-Image-SFT/dit_3.5bit.jpg"></td>
 
50
  </tr>
51
  </table>
52
 
 
54
 
55
  <table width="100%">
56
  <tr>
57
+ <td width="33%" align="center">bf16 (reference)</td>
58
+ <td width="33%" align="center">6bit</td>
59
+ <td width="33%" align="center">int8@7</td>
 
60
  </tr>
61
  <tr>
62
  <td align="center"><img width="100%" src="assets/MOSS-Image-Base/bf16.jpg"></td>
63
  <td align="center"><img width="100%" src="assets/MOSS-Image-Base/dit_6bit.jpg"></td>
64
+ <td align="center"><img width="100%" src="assets/MOSS-Image-Base/dit_int8_7bit.jpg"></td>
65
+ </tr>
66
+ <tr>
67
+ <td width="33%" align="center">int8@6</td>
68
+ <td width="33%" align="center">4.5bit</td>
69
+ <td></td>
70
+ </tr>
71
+ <tr>
72
  <td align="center"><img width="100%" src="assets/MOSS-Image-Base/dit_int8_6bit.jpg"></td>
73
+ <td align="center"><img width="100%" src="assets/MOSS-Image-Base/dit_4.5bit.jpg"></td>
74
+ <td></td>
75
  </tr>
76
  </table>
77
 
 
84
  | dit_3.5bit (mixed) | `models/MOSS-Image-SFT/dit_3.5bit.safetensors` | 3.526 | 2.24 GB |
85
  | dit_4.5bit (mixed) | `models/MOSS-Image-SFT/dit_4.5bit.safetensors` | 4.545 | 2.89 GB |
86
  | dit_6bit | `models/MOSS-Image-SFT/dit_6bit.safetensors` | 6.055 | 3.85 GB |
87
+ | dit_int8_6bit (hybrid INT8 GEMM + ConvRot) | `models/MOSS-Image-SFT/dit_int8_6bit.safetensors` | 5.999 | 3.81 GB |
88
+ | dit_int8_7bit (hybrid INT8 GEMM + ConvRot) | `models/MOSS-Image-SFT/dit_int8_7bit.safetensors` | 7.004 | 4.45 GB |
89
 
90
  ### Base
91
 
 
93
  |---|---|---|---|
94
  | dit_4.5bit (mixed) | `models/MOSS-Image-Base/dit_4.5bit.safetensors` | 4.545 | 2.89 GB |
95
  | dit_6bit | `models/MOSS-Image-Base/dit_6bit.safetensors` | 6.055 | 3.85 GB |
96
+ | dit_int8_6bit (hybrid INT8 GEMM + ConvRot) | `models/MOSS-Image-Base/dit_int8_6bit.safetensors` | 5.999 | 3.81 GB |
97
+ | dit_int8_7bit (hybrid INT8 GEMM + ConvRot) | `models/MOSS-Image-Base/dit_int8_7bit.safetensors` | 7.004 | 4.45 GB |
98
 
99
  ## Text Encoder
100
 
 
104
  |---|---|---|---|
105
  | text_encoder_4.5bit | `models/text_encoder_4.5bit.safetensors` | 4.552 | 5.97 GB |
106
  | text_encoder_6bit | `models/text_encoder_6bit.safetensors` | 6.070 | 7.29 GB |
107
+ | text_encoder_int8_6bit (hybrid INT8 GEMM + ConvRot) | `models/text_encoder_int8_6bit.safetensors` | 6.000 | 7.23 GB |
108
+ | text_encoder_int8_7bit (hybrid INT8 GEMM + ConvRot) | `models/text_encoder_int8_7bit.safetensors` | 6.993 | 8.08 GB |
109
 
110
  ## Usage
111
 
 
123
 
124
  ### Example scripts
125
 
126
+ * Inference: [standard](https://github.com/modelscope/DiffSynth-Studio/blob/main/examples/moss_image/model_inference/MOSS-Image-EntroPack.py) / [low VRAM](https://github.com/modelscope/DiffSynth-Studio/blob/main/examples/moss_image/model_inference_low_vram/MOSS-Image-EntroPack.py) — defaults to SFT INT8 6bit; switch `dit_package` / `base_model_id` together to use Base or another level
127
  * LoRA training: [train](https://github.com/modelscope/DiffSynth-Studio/blob/main/examples/moss_image/model_training/lora/MOSS-Image-entropack.sh) / [validation](https://github.com/modelscope/DiffSynth-Studio/blob/main/examples/moss_image/model_training/validate_lora/MOSS-Image-entropack.py) — defaults to SFT; replace `MOSS-Image-SFT` with `MOSS-Image-Base` in the dataset and model paths to train on Base
128
 
129
+ `dit_package` and `text_encoder_package` are chosen independently: any combination works (e.g. `dit_int8_6bit` with `text_encoder_4.5bit`). The 6-bit packages are pre-registered in DiffSynth-Studio, so they load with a plain `ModelConfig`; the helper below registers any package's json sidecar before loading (the INT8 packages also read their ConvRot settings and 8.5-bit sensitive-layer allocation from it):
130
 
131
  ```python
132
  import json
 
166
  )
167
 
168
  image = pipe(
169
+ prompt="""Full-color cinematic outdoor night photograph of a stunning young Asian woman in her early twenties with a natural warm ivory complexion and a healthy glow, wearing a soft blush pink off-shoulder dress with a fitted bodice, framed from the waist up against softly illuminated historic architecture. She stands near the centre of the frame, facing the camera with a gentle closed-lip smile, her hands resting loosely clasped in front of her waist with relaxed elbows; she has a slim oval face with softly defined cheekbones and a delicate narrow jaw, dark expressive eyes, natural-shaped brows, and subtle natural makeup with soft red lips. Her long black hair falls in loose natural waves over her shoulders, with fine strands framing her face. The fabric of the dress falls in soft folds across her shoulders and chest. In the frame-left background, the word "EntroPack" glows as a hand-drawn bubbly yellow neon sign mounted on a softly floodlit dark stone building above the street; rows of warm-lit windows, stone cornices and shopfronts remain clearly readable around the sign in gentle amber light, while the thick rounded irregular letters ride a gently bouncing baseline with a bright core and a soft outer glow, kept sharp and unblurred against the night. The background recedes into deep blue tones: the Eiffel Tower rises above distant buildings on the frame-right as a softly silhouetted tapering lattice outline in the shallow depth of field, while small amber, muted turquoise and rose-coloured lights dissolve into rounded bokeh across the upper background, leaving a darker area behind her head. Soft neutral frontal illumination with gentle falloff reveals her face with natural warm skin tone and subtle texture, a faint cooler rim separates her hair from the night, and the skirt of her dress fills the lower edge of the frame.""",
170
  height=1024, width=1024,
171
+ seed=123, num_inference_steps=50, cfg_scale=3.0,
172
  )
173
  image.save("image.jpg")
174
  ```
assets/MOSS-Image-Base/dit_int8_6bit.jpg CHANGED

Git LFS Details

  • SHA256: b511902d1ad5146aca0619ffc9e217a066fe5e2e12e76b91ca7ca5270a5341b5
  • Pointer size: 131 Bytes
  • Size of remote file: 104 kB

Git LFS Details

  • SHA256: 9893fdede20e42f6649882f4cb669ed8b000c63e27a4ed7fc9adb472aa92a6d9
  • Pointer size: 130 Bytes
  • Size of remote file: 93 kB
assets/MOSS-Image-Base/dit_int8_7bit.jpg ADDED
assets/MOSS-Image-SFT/dit_int8_6bit.jpg CHANGED

Git LFS Details

  • SHA256: c49d56b9a00d11a6f181aa5c97dfec29865292a8f8500dc90d5e0f0fca95513e
  • Pointer size: 131 Bytes
  • Size of remote file: 111 kB

Git LFS Details

  • SHA256: 3f9893dbd40bdc3573f618829de8fc3db2d99e72cbd892e59ec49392c83c69a6
  • Pointer size: 131 Bytes
  • Size of remote file: 121 kB
assets/MOSS-Image-SFT/dit_int8_7bit.jpg ADDED

Git LFS Details

  • SHA256: 2046618eccfa1c6899250ffac83c19483740894b6c78c08678ddcc7a9396804d
  • Pointer size: 131 Bytes
  • Size of remote file: 111 kB
models/MOSS-Image-Base/dit_int8_6bit.json CHANGED
@@ -1,14 +1,66 @@
1
  {
2
- "model_hash": "36922e1c521a9873c7d507512d25ab2e",
3
  "model_name": "moss_image_dit",
4
  "model_class": "diffsynth.models.moss_image_dit.MOSSImageDiT",
5
  "quant_config": {
6
- "method": "entropack_lossy_quant_int8",
7
  "load_prequantized": true,
8
- "backend_config_kwargs": {
9
- "target_bpp": 6.0,
10
- "row_rdo_iterations": 2,
11
- "row_rdo_candidates": 20
12
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  }
14
  }
 
1
  {
2
+ "model_hash": "955b29367fb6e506f5fb6b3acd50c1ed",
3
  "model_name": "moss_image_dit",
4
  "model_class": "diffsynth.models.moss_image_dit.MOSSImageDiT",
5
  "quant_config": {
 
6
  "load_prequantized": true,
7
+ "configs": [
8
+ {
9
+ "method": "entropack_lossy_quant_int8",
10
+ "backend_config_kwargs": {
11
+ "target_bpp": 5.782,
12
+ "row_rdo_iterations": 2,
13
+ "row_rdo_candidates": 20,
14
+ "use_convrot": true,
15
+ "convrot_groupsize": 256
16
+ },
17
+ "exclude_modules": [
18
+ "video_embedder",
19
+ "text_embedder.proj",
20
+ "text_embedder.blocks.0.qkv",
21
+ "text_embedder.blocks.0.proj",
22
+ "text_embedder.blocks.0.ffn_gate",
23
+ "text_embedder.blocks.0.ffn_up",
24
+ "text_embedder.blocks.0.ffn_down",
25
+ "text_embedder.blocks.1.qkv",
26
+ "text_embedder.blocks.1.proj",
27
+ "text_embedder.blocks.1.ffn_gate",
28
+ "text_embedder.blocks.1.ffn_up",
29
+ "text_embedder.blocks.1.ffn_down",
30
+ "time_embedding.0",
31
+ "time_embedding.2",
32
+ "shared_block_modulation",
33
+ "final_linear_video",
34
+ "video_head_modulation"
35
+ ]
36
+ },
37
+ {
38
+ "method": "entropack_lossy_quant",
39
+ "backend_config_kwargs": {
40
+ "target_bpp": 8.5,
41
+ "row_rdo_iterations": 2,
42
+ "row_rdo_candidates": 20
43
+ },
44
+ "target_modules": [
45
+ "video_embedder",
46
+ "text_embedder.proj",
47
+ "text_embedder.blocks.0.qkv",
48
+ "text_embedder.blocks.0.proj",
49
+ "text_embedder.blocks.0.ffn_gate",
50
+ "text_embedder.blocks.0.ffn_up",
51
+ "text_embedder.blocks.0.ffn_down",
52
+ "text_embedder.blocks.1.qkv",
53
+ "text_embedder.blocks.1.proj",
54
+ "text_embedder.blocks.1.ffn_gate",
55
+ "text_embedder.blocks.1.ffn_up",
56
+ "text_embedder.blocks.1.ffn_down",
57
+ "time_embedding.0",
58
+ "time_embedding.2",
59
+ "shared_block_modulation",
60
+ "final_linear_video",
61
+ "video_head_modulation"
62
+ ]
63
+ }
64
+ ]
65
  }
66
  }
models/MOSS-Image-Base/dit_int8_6bit.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3ebb384ba6527dd10ce7d7eeaaeb8a94718b58cd99d3a9f59f9e61d9ca0d5e13
3
- size 3852165592
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b94917cab4f43b56d5356b5bfbea747d1ff0910390f01aef1cca1434cdd211e9
3
+ size 3811723984
models/MOSS-Image-Base/dit_int8_7bit.json ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_hash": "784655dc0c006e3ef6a3e89fa2975f7e",
3
+ "model_name": "moss_image_dit",
4
+ "model_class": "diffsynth.models.moss_image_dit.MOSSImageDiT",
5
+ "quant_config": {
6
+ "load_prequantized": true,
7
+ "configs": [
8
+ {
9
+ "method": "entropack_lossy_quant_int8",
10
+ "backend_config_kwargs": {
11
+ "target_bpp": 6.854,
12
+ "row_rdo_iterations": 2,
13
+ "row_rdo_candidates": 20,
14
+ "use_convrot": true,
15
+ "convrot_groupsize": 256
16
+ },
17
+ "exclude_modules": [
18
+ "video_embedder",
19
+ "text_embedder.proj",
20
+ "text_embedder.blocks.0.qkv",
21
+ "text_embedder.blocks.0.proj",
22
+ "text_embedder.blocks.0.ffn_gate",
23
+ "text_embedder.blocks.0.ffn_up",
24
+ "text_embedder.blocks.0.ffn_down",
25
+ "text_embedder.blocks.1.qkv",
26
+ "text_embedder.blocks.1.proj",
27
+ "text_embedder.blocks.1.ffn_gate",
28
+ "text_embedder.blocks.1.ffn_up",
29
+ "text_embedder.blocks.1.ffn_down",
30
+ "time_embedding.0",
31
+ "time_embedding.2",
32
+ "shared_block_modulation",
33
+ "final_linear_video",
34
+ "video_head_modulation"
35
+ ]
36
+ },
37
+ {
38
+ "method": "entropack_lossy_quant",
39
+ "backend_config_kwargs": {
40
+ "target_bpp": 8.5,
41
+ "row_rdo_iterations": 2,
42
+ "row_rdo_candidates": 20
43
+ },
44
+ "target_modules": [
45
+ "video_embedder",
46
+ "text_embedder.proj",
47
+ "text_embedder.blocks.0.qkv",
48
+ "text_embedder.blocks.0.proj",
49
+ "text_embedder.blocks.0.ffn_gate",
50
+ "text_embedder.blocks.0.ffn_up",
51
+ "text_embedder.blocks.0.ffn_down",
52
+ "text_embedder.blocks.1.qkv",
53
+ "text_embedder.blocks.1.proj",
54
+ "text_embedder.blocks.1.ffn_gate",
55
+ "text_embedder.blocks.1.ffn_up",
56
+ "text_embedder.blocks.1.ffn_down",
57
+ "time_embedding.0",
58
+ "time_embedding.2",
59
+ "shared_block_modulation",
60
+ "final_linear_video",
61
+ "video_head_modulation"
62
+ ]
63
+ }
64
+ ]
65
+ }
66
+ }
models/MOSS-Image-Base/dit_int8_7bit.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:225c783fcd6a0f0af10ad90771942c5788e7b4c07b6e985cab4e0637131d2e62
3
+ size 4449823136
models/MOSS-Image-SFT/dit_int8_6bit.json CHANGED
@@ -1,14 +1,66 @@
1
  {
2
- "model_hash": "ad5f9335f74b2eb9d439ed118b0625ac",
3
  "model_name": "moss_image_dit",
4
  "model_class": "diffsynth.models.moss_image_dit.MOSSImageDiT",
5
  "quant_config": {
6
- "method": "entropack_lossy_quant_int8",
7
  "load_prequantized": true,
8
- "backend_config_kwargs": {
9
- "target_bpp": 6.0,
10
- "row_rdo_iterations": 2,
11
- "row_rdo_candidates": 20
12
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  }
14
  }
 
1
  {
2
+ "model_hash": "060367613bb0ebec4631f137437bfeb8",
3
  "model_name": "moss_image_dit",
4
  "model_class": "diffsynth.models.moss_image_dit.MOSSImageDiT",
5
  "quant_config": {
 
6
  "load_prequantized": true,
7
+ "configs": [
8
+ {
9
+ "method": "entropack_lossy_quant_int8",
10
+ "backend_config_kwargs": {
11
+ "target_bpp": 5.782,
12
+ "row_rdo_iterations": 2,
13
+ "row_rdo_candidates": 20,
14
+ "use_convrot": true,
15
+ "convrot_groupsize": 256
16
+ },
17
+ "exclude_modules": [
18
+ "video_embedder",
19
+ "text_embedder.proj",
20
+ "text_embedder.blocks.0.qkv",
21
+ "text_embedder.blocks.0.proj",
22
+ "text_embedder.blocks.0.ffn_gate",
23
+ "text_embedder.blocks.0.ffn_up",
24
+ "text_embedder.blocks.0.ffn_down",
25
+ "text_embedder.blocks.1.qkv",
26
+ "text_embedder.blocks.1.proj",
27
+ "text_embedder.blocks.1.ffn_gate",
28
+ "text_embedder.blocks.1.ffn_up",
29
+ "text_embedder.blocks.1.ffn_down",
30
+ "time_embedding.0",
31
+ "time_embedding.2",
32
+ "shared_block_modulation",
33
+ "final_linear_video",
34
+ "video_head_modulation"
35
+ ]
36
+ },
37
+ {
38
+ "method": "entropack_lossy_quant",
39
+ "backend_config_kwargs": {
40
+ "target_bpp": 8.5,
41
+ "row_rdo_iterations": 2,
42
+ "row_rdo_candidates": 20
43
+ },
44
+ "target_modules": [
45
+ "video_embedder",
46
+ "text_embedder.proj",
47
+ "text_embedder.blocks.0.qkv",
48
+ "text_embedder.blocks.0.proj",
49
+ "text_embedder.blocks.0.ffn_gate",
50
+ "text_embedder.blocks.0.ffn_up",
51
+ "text_embedder.blocks.0.ffn_down",
52
+ "text_embedder.blocks.1.qkv",
53
+ "text_embedder.blocks.1.proj",
54
+ "text_embedder.blocks.1.ffn_gate",
55
+ "text_embedder.blocks.1.ffn_up",
56
+ "text_embedder.blocks.1.ffn_down",
57
+ "time_embedding.0",
58
+ "time_embedding.2",
59
+ "shared_block_modulation",
60
+ "final_linear_video",
61
+ "video_head_modulation"
62
+ ]
63
+ }
64
+ ]
65
  }
66
  }
models/MOSS-Image-SFT/dit_int8_6bit.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1c16c9bde987116e76d042ba1b6a70fff633bdb5ba402aeaf360191e6184d7ca
3
- size 3852155752
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ac2bb3ec2b2f02eb56e71aaabe91faf5cedfae138fd9403f16ac980e9672b44
3
+ size 3811724464
models/MOSS-Image-SFT/dit_int8_7bit.json ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_hash": "2a95c50f3b178e8f49404604c1b8426e",
3
+ "model_name": "moss_image_dit",
4
+ "model_class": "diffsynth.models.moss_image_dit.MOSSImageDiT",
5
+ "quant_config": {
6
+ "load_prequantized": true,
7
+ "configs": [
8
+ {
9
+ "method": "entropack_lossy_quant_int8",
10
+ "backend_config_kwargs": {
11
+ "target_bpp": 6.854,
12
+ "row_rdo_iterations": 2,
13
+ "row_rdo_candidates": 20,
14
+ "use_convrot": true,
15
+ "convrot_groupsize": 256
16
+ },
17
+ "exclude_modules": [
18
+ "video_embedder",
19
+ "text_embedder.proj",
20
+ "text_embedder.blocks.0.qkv",
21
+ "text_embedder.blocks.0.proj",
22
+ "text_embedder.blocks.0.ffn_gate",
23
+ "text_embedder.blocks.0.ffn_up",
24
+ "text_embedder.blocks.0.ffn_down",
25
+ "text_embedder.blocks.1.qkv",
26
+ "text_embedder.blocks.1.proj",
27
+ "text_embedder.blocks.1.ffn_gate",
28
+ "text_embedder.blocks.1.ffn_up",
29
+ "text_embedder.blocks.1.ffn_down",
30
+ "time_embedding.0",
31
+ "time_embedding.2",
32
+ "shared_block_modulation",
33
+ "final_linear_video",
34
+ "video_head_modulation"
35
+ ]
36
+ },
37
+ {
38
+ "method": "entropack_lossy_quant",
39
+ "backend_config_kwargs": {
40
+ "target_bpp": 8.5,
41
+ "row_rdo_iterations": 2,
42
+ "row_rdo_candidates": 20
43
+ },
44
+ "target_modules": [
45
+ "video_embedder",
46
+ "text_embedder.proj",
47
+ "text_embedder.blocks.0.qkv",
48
+ "text_embedder.blocks.0.proj",
49
+ "text_embedder.blocks.0.ffn_gate",
50
+ "text_embedder.blocks.0.ffn_up",
51
+ "text_embedder.blocks.0.ffn_down",
52
+ "text_embedder.blocks.1.qkv",
53
+ "text_embedder.blocks.1.proj",
54
+ "text_embedder.blocks.1.ffn_gate",
55
+ "text_embedder.blocks.1.ffn_up",
56
+ "text_embedder.blocks.1.ffn_down",
57
+ "time_embedding.0",
58
+ "time_embedding.2",
59
+ "shared_block_modulation",
60
+ "final_linear_video",
61
+ "video_head_modulation"
62
+ ]
63
+ }
64
+ ]
65
+ }
66
+ }
models/MOSS-Image-SFT/dit_int8_7bit.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a8838b21cd89d5958fb4b8424d1cb01cddedb328ef69f95c0d3ef46751dea2a
3
+ size 4449793120
models/text_encoder_int8_6bit.json CHANGED
@@ -1,14 +1,36 @@
1
  {
2
- "model_hash": "003277dd4ed448237a756b6502a65d57",
3
  "model_name": "moss_image_text_encoder",
4
  "model_class": "diffsynth.models.moss_image_text_encoder.MOSSImageTextEncoder",
5
  "quant_config": {
6
- "method": "entropack_lossy_quant_int8",
7
  "load_prequantized": true,
8
- "backend_config_kwargs": {
9
- "target_bpp": 6.0,
10
- "row_rdo_iterations": 2,
11
- "row_rdo_candidates": 20
12
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  }
14
  }
 
1
  {
2
+ "model_hash": "08ba3e9e8267bc600b0285356c2d5b86",
3
  "model_name": "moss_image_text_encoder",
4
  "model_class": "diffsynth.models.moss_image_text_encoder.MOSSImageTextEncoder",
5
  "quant_config": {
 
6
  "load_prequantized": true,
7
+ "configs": [
8
+ {
9
+ "method": "entropack_lossy_quant_int8",
10
+ "backend_config_kwargs": {
11
+ "target_bpp": 5.956,
12
+ "row_rdo_iterations": 2,
13
+ "row_rdo_candidates": 20,
14
+ "use_convrot": true,
15
+ "convrot_groupsize": 256
16
+ },
17
+ "exclude_modules": [
18
+ "linear_attn.in_proj_a",
19
+ "linear_attn.in_proj_b"
20
+ ]
21
+ },
22
+ {
23
+ "method": "entropack_lossy_quant",
24
+ "backend_config_kwargs": {
25
+ "target_bpp": 8.5,
26
+ "row_rdo_iterations": 2,
27
+ "row_rdo_candidates": 20
28
+ },
29
+ "target_modules": [
30
+ "linear_attn.in_proj_a",
31
+ "linear_attn.in_proj_b"
32
+ ]
33
+ }
34
+ ]
35
  }
36
  }
models/text_encoder_int8_6bit.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:aa98d23b48c3af573d441cc027262ec7656759596b671c7d44ed6d469a0ea4d9
3
- size 7290452872
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:040ddf9ab3df1c1363498ea73b67b4dc979a7d843404266e3ad35b0bdc5c2b24
3
+ size 7225461096
models/text_encoder_int8_7bit.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_hash": "1f0b627389760b03f597e10af0e1f4f6",
3
+ "model_name": "moss_image_text_encoder",
4
+ "model_class": "diffsynth.models.moss_image_text_encoder.MOSSImageTextEncoder",
5
+ "quant_config": {
6
+ "load_prequantized": true,
7
+ "configs": [
8
+ {
9
+ "method": "entropack_lossy_quant_int8",
10
+ "backend_config_kwargs": {
11
+ "target_bpp": 6.951,
12
+ "row_rdo_iterations": 2,
13
+ "row_rdo_candidates": 20,
14
+ "use_convrot": true,
15
+ "convrot_groupsize": 256
16
+ },
17
+ "exclude_modules": [
18
+ "linear_attn.in_proj_a",
19
+ "linear_attn.in_proj_b"
20
+ ]
21
+ },
22
+ {
23
+ "method": "entropack_lossy_quant",
24
+ "backend_config_kwargs": {
25
+ "target_bpp": 8.5,
26
+ "row_rdo_iterations": 2,
27
+ "row_rdo_candidates": 20
28
+ },
29
+ "target_modules": [
30
+ "linear_attn.in_proj_a",
31
+ "linear_attn.in_proj_b"
32
+ ]
33
+ }
34
+ ]
35
+ }
36
+ }
models/text_encoder_int8_7bit.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf3ee3ff11c380c1ea246f2ed5f06a73cd075686f303ac7a4abe75f3414936c4
3
+ size 8084231096