Goekdeniz-Guelmez 710bc1490e training mode working too got from 2 toks/sec to 30 toks/sec with raw 1.5B model

2025-02-21 22:42:15 +01:00

..

__init__.py

LoRA: Extract small function (#614)

2024-06-02 06:38:42 -07:00

datasets.py

Huge speed improvement in validation mode.

2025-02-21 22:08:49 +01:00

dora.py

Feature: QDoRA (#891)

2024-09-30 08:01:11 -07:00

grpo_trainer.py

training mode working too got from 2 toks/sec to 30 toks/sec with raw 1.5B model

2025-02-21 22:42:15 +01:00

lora.py

Allow the entire model to be targed for LoRA and DoRA fine tuning: LoRA and DoRA embeddings with small DoRALinear bug fix (#914)

2024-08-16 07:38:36 -07:00

trainer.py

fix lora timings after validation (#1278)

2025-02-11 16:48:55 -08:00

utils.py

Merge branch 'ml-explore:main' into adding-GRPO-training

2025-02-12 11:10:10 +01:00