kernelpool commited on
Commit
ba6849b
·
verified ·
1 Parent(s): 50438e2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +25 -1
README.md CHANGED
@@ -4,14 +4,38 @@ library_name: mlx
4
  pipeline_tag: text-generation
5
  tags:
6
  - mlx
 
 
7
  ---
8
 
9
  # kernelpool/LongCat-2.0-3bit-UVMAX
10
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  ## Use with mlx
12
 
 
 
 
 
13
  ```bash
14
- pip install mlx-lm
15
  ```
16
 
17
  ```python
 
4
  pipeline_tag: text-generation
5
  tags:
6
  - mlx
7
+ license: mit
8
+ base_model: meituan-longcat/LongCat-2.0-FP8
9
  ---
10
 
11
  # kernelpool/LongCat-2.0-3bit-UVMAX
12
 
13
+ Mixed-precision (UVMAX) quantization of [meituan-longcat/LongCat-2.0-FP8](https://huggingface.co/meituan-longcat/LongCat-2.0-FP8),
14
+ converted with [mlx-lm](https://github.com/ml-explore/mlx-lm).
15
+
16
+ ## What is UVMAX?
17
+
18
+ UVMAX is a mixed-precision scheme: bit widths are assigned per tensor class
19
+ from measured round-trip quantization error, rather than uniformly. All
20
+ classes use group size 64.
21
+
22
+ | Tensor class | Bits | Parameters | Size | Share |
23
+ |---|---|---|---|---|
24
+ | Expert FFNs | 3 | 1.47 T | 598.5 GiB | 88.0% |
25
+ | N-gram embedding tables | 3 | 135 B | 55.0 GiB | 8.1% |
26
+ | Attention, dense MLPs | 6 | 31.4 B | 22.9 GiB | 3.4% |
27
+ | Embeddings, `lm_head` | 6 | 2.8 B | 2.1 GiB | 0.3% |
28
+ | DSA indexer, MoE routers | 8 | 0.6 B | 0.6 GiB | 0.1% |
29
+ | Norms, correction biases (unquantized) | — | — | 0.9 GiB | 0.1% |
30
+
31
  ## Use with mlx
32
 
33
+ This model requires LongCat-2.0 support from [mlx-lm PR #1464](https://github.com/ml-explore/mlx-lm/pull/1464),
34
+ which has not yet been merged. Until it is included in an mlx-lm release, install
35
+ mlx-lm from the PR branch:
36
+
37
  ```bash
38
+ pip install git+https://github.com/ml-explore/mlx-lm.git@refs/pull/1464/head
39
  ```
40
 
41
  ```python