Commit Graph

12 Commits

Author SHA1 Message Date
Christian Medina
7040c84618 feat: process 2 shards at a time (one per GPU) 2026-07-02 23:05:57 -04:00
Christian Medina
d698a32f0f feat: skip already processed shards (resume support) 2026-07-02 23:04:52 -04:00
Christian Medina
d1f0c81ac4 fix: remove quant_states reference 2026-07-02 22:52:49 -04:00
Christian Medina
18ef0f73ef fix: don't save quant_state objects (BnB rebuilds on load) 2026-07-02 22:48:09 -04:00
Christian Medina
8c13be4d1f fix: free GPU memory after each layer quantization 2026-07-02 22:43:27 -04:00
Christian Medina
0f624e1e75 fix: use bitsandbytes.functional for expert layer quantization 2026-07-02 22:36:02 -04:00
Christian Medina
bfa569f4e9 fix: remove streaming from output path name 2026-07-02 22:06:50 -04:00
Christian Medina
959c43d44c feat: true streaming NF4 quantization with safetensors 2026-07-02 22:05:59 -04:00
Christian Medina
d41648d4fc feat: load shards directly to GPU, no CPU bottleneck 2026-07-02 21:44:28 -04:00
Christian Medina
82677070ff fix: weights_only=False for safetensors loading 2026-07-02 21:41:44 -04:00
Christian Medina
aec7f405fe feat: use both GPUs in parallel for streaming quantization 2026-07-02 21:30:35 -04:00
Christian Medina
0ea51488a7 feat: add streaming quantization script (process one shard at a time) 2026-07-02 21:28:05 -04:00