Goekdeniz-Guelmez 05d921b788 optims

2025-02-03 19:37:05 +01:00

..

__init__.py

LoRA: Extract small function (#614)

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

datasets.py

first working prototype, will try training out at home

2025-02-03 12:05:29 +01:00

dora.py

Feature: QDoRA (#891)

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

grpo_trainer.py

optims

2025-02-03 19:37:05 +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

reduction moved to CPU in case of distributed training (#1200)

2025-01-14 17:20:42 -08:00

utils.py

adding support for kyutai's helium (#1208)

2025-01-26 07:19:07 -08:00