Commit Graph

23 Commits

Author SHA1 Message Date
Christian Medina
0752c03b83 fix: use absolute path for dataset (no ~ expansion) 2026-07-02 16:04:21 -04:00
Christian Medina
b8214f64cb fix: use 4-bit model, remove DeepSpeed, fix dataset path 2026-07-02 15:23:26 -04:00
Christian Medina
8747111671 fix: update config comment to reflect Ornith-35B 2026-07-02 00:11:10 -04:00
Christian Medina
a655040db2 feat: use DeepSpeed ZeRO-3 for proper model sharding across GPUs 2026-07-01 22:08:37 -04:00
Christian Medina
815115e97a feat: add BnB 4-bit quantization strategy + use bf16 model 2026-07-01 21:14:31 -04:00
Christian Medina
3760aaf316 fix: rename output dir to ornith-35b-lora 2026-07-01 14:19:43 -04:00
Christian Medina
14042dbb44 fix: use pre-quantized 4bit model 2026-07-01 12:35:19 -04:00
Christian Medina
2fa5173c56 feat: add prepare_model_for_kbit_training and update LoRA r=64 alpha=128 2026-07-01 07:48:44 -04:00
Christian Medina
da5eb3abed feat: implement QLoRA with 4-bit BitsAndBytes quantization 2026-07-01 07:45:33 -04:00
Christian Medina
f47944ee75 fix: remove quantization_config after loading to bypass SFTTrainer check 2026-07-01 07:36:38 -04:00
Christian Medina
6f631f00ae fix: use FP8 model with 32GB CPU offload 2026-07-01 07:29:32 -04:00
Christian Medina
9d92531c55 fix: use Ornith-1.0-35B bf16 base model (PEFT compatible) 2026-07-01 07:28:21 -04:00
Christian Medina
8d4cdca3bb feat: use Ornith-1.0-35B-NVFP4 for training 2026-06-30 23:34:33 -04:00
Christian Medina
5713bf0285 fix: add CPU offload back to DeepSpeed ZeRO-3 2026-06-30 22:47:19 -04:00
Christian Medina
a8c9cd682e fix: simplify DeepSpeed config, remove offload settings 2026-06-30 22:36:49 -04:00
Christian Medina
6bd6662f5f docs: rename config and docs to reflect Ornith-35B model 2026-06-30 20:41:53 -04:00
Christian Medina
50fa171f68 fix: use local model path instead of HuggingFace download 2026-06-30 20:24:44 -04:00
Christian Medina
97004261f5 fix: multiple bug fixes - remove unused imports, fix typo, fix dataset paths, add ai command fallback 2026-06-30 18:48:46 -04:00
Christian Medina
4f2f8ef03f feat: use DeepSpeed ZeRO-3 for optimal 2-GPU training 2026-06-30 18:26:48 -04:00
Christian Medina
cd58ffd31e fix: use FSDP for 2-GPU training, reduce batch size 2026-06-30 18:23:34 -04:00
Christian Medina
ef42e67183 fix: use CompressedTensors for Ornith, add torch import, expand LoRA targets 2026-06-30 17:54:35 -04:00
Christian Medina
2074d3d993 feat: use Ornith-1.0-35B-FP8 with multi-GPU support 2026-06-30 15:57:54 -04:00
Christian Medina
418a4cc76d init: add LoRA training infrastructure and 20k dataset 2026-06-30 14:49:44 -04:00