Commit Graph

216 Commits

Author SHA1 Message Date
Christian Medina
49b99889ef feat: add proper BnB quantization script (loads to CPU, saves) 2026-07-03 01:39:27 -04:00
Christian Medina
b82be559dc fix: use Linear4bit to preserve weight shapes during quantization 2026-07-03 01:38:54 -04:00
Christian Medina
aa7961605f fix: add index file creation for sharded models 2026-07-03 01:25:18 -04:00
Christian Medina
5d0ff6c8af simplify: only train, no quantization logic 2026-07-03 01:25:11 -04:00
Christian Medina
6f45390be8 fix: remove quantization_config from config.json to prevent re-quantization 2026-07-03 01:24:25 -04:00
Christian Medina
1c3da52028 feat: create proper index file from safetensors metadata 2026-07-03 01:22:54 -04:00
Christian Medina
b15bc5c975 feat: detect pre-quantized model and skip re-quantization 2026-07-03 01:19:55 -04:00
Christian Medina
4308fd3391 feat: continuous processing - add to GPU with fewer tasks (max 4 per GPU) 2026-07-03 00:26:38 -04:00
Christian Medina
8a30ec92c6 fix: reduce to 3 shards per GPU (6 total) for memory 2026-07-03 00:23:21 -04:00
Christian Medina
23dcdd2c6f feat: rename 0-indexed saved files to 1-indexed 2026-07-03 00:16:15 -04:00
Christian Medina
b4538c013b fix: use 1-indexing for shard names (model-00001-of-00016) 2026-07-03 00:15:39 -04:00
Christian Medina
50623cdcd0 fix: use 0-indexing consistently (shard 0-15) 2026-07-03 00:14:57 -04:00
Christian Medina
262b1181b7 fix: skip to first unsaved shard, fix GPU assignment 2026-07-02 23:39:24 -04:00
Christian Medina
b742b5549e fix: alternate GPUs for each shard (0,1,0,1) 2026-07-02 23:32:38 -04:00
Christian Medina
78817532ad fix: import concurrent.futures at top level 2026-07-02 23:27:14 -04:00
Christian Medina
bc3bc6d6eb feat: process 8 shards at once (4 per GPU) with better logging 2026-07-02 23:24:48 -04:00
Christian Medina
9e2c7bec8d fix: remove duplicate sequential code 2026-07-02 23:12:03 -04:00
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
b9f6ddaf50 fix: free GPU memory after each quantization 2026-07-02 22:41:52 -04:00
Christian Medina
1751d33f69 fix: load shard to CPU, quantize on GPU 2026-07-02 22:40:31 -04:00
Christian Medina
d769d39865 feat: add single shard test script 2026-07-02 22:39:25 -04:00
Christian Medina
0f624e1e75 fix: use bitsandbytes.functional for expert layer quantization 2026-07-02 22:36:02 -04:00
Christian Medina
410718bc73 feat: add CPU offloading for 70GB → 17GB conversion 2026-07-02 22:33:36 -04:00
Christian Medina
bee6683396 feat: simple NF4 quantization with device_map=auto (proven method) 2026-07-02 22:32:38 -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
52b93837f9 fix: use NF4 naming convention for quantized model 2026-07-02 22:00:15 -04:00
Christian Medina
2f997377f5 feat: add simple BnB 4-bit quantization script with device_map=auto 2026-07-02 21:57:02 -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
Christian Medina
170243de25 fix: remove GPU test, just check quantization 2026-07-02 21:14:34 -04:00
Christian Medina
43f5426583 feat: test BnB 4-bit loading and GPU transfer 2026-07-02 21:10:57 -04:00
Christian Medina
04a95f362f fix: load bf16 model directly with BnB 4-bit (single load) 2026-07-02 21:09:54 -04:00
Christian Medina
1f804c95cd feat: load bf16 to CPU, quantize with BnB, move to GPU 2026-07-02 20:58:22 -04:00
Christian Medina
a9b066246d feat: add model size check script 2026-07-02 20:47:11 -04:00
Christian Medina
fa2f21562f fix: load BnB 4-bit model to single GPU (no distribution) 2026-07-02 20:42:05 -04:00
Christian Medina
a23ecc49f0 feat: use transformers built-in BnB loading with device_map=cpu 2026-07-02 20:27:06 -04:00
Christian Medina
11f9c3a56c fix: force quantization with dummy forward pass 2026-07-02 20:25:26 -04:00
Christian Medina
b624e90b02 fix: use BnB quantize_batch to actually quantize weights 2026-07-02 20:22:39 -04:00
Christian Medina
ec2b1be864 chore: clean up temp files 2026-07-02 20:18:13 -04:00
Christian Medina
3716051abc fix: remove PEFT prepare (causes OOM) 2026-07-02 20:17:23 -04:00
Christian Medina
ac891fe25b fix: replace with proper BnB 4-bit quantization script 2026-07-02 20:13:19 -04:00
Christian Medina
7984ae93e7 feat: proper BnB 4-bit quantization with PEFT prepare 2026-07-02 20:11:38 -04:00
Christian Medina
57b169f202 feat: use BnB 4-bit model for QLoRA training 2026-07-02 20:04:55 -04:00