Code

TRL SFTTrainer Examples#2211

Code

Merged

bfineran merged 24 commits intomainneuralmagic/sparseml:mainfrom sa/sft_trainer_mixinneuralmagic/sparseml:sa/sft_trainer_mixinCopy head branch name to clipboardApr 24, 2024

+276-149

Commits

Commits on Mar 5, 2024

• WIP sft mixin

  

  Sara Adkins

  committed
  f013c2c

  Copy full SHA for f013c2c

• its running at least

  

  Sara Adkins

  committed
  9986f34

  Copy full SHA for 9986f34

• clean up

  

  Sara Adkins

  committed
  51dd109

  Copy full SHA for 51dd109

Commits on Apr 1, 2024

• Merge branch 'main' into sa/sft_trainer_mixin

  

  Sara Adkins

  committed
  72e3685

  Copy full SHA for 72e3685

• revert debugging changes

  

  Sara Adkins

  committed
  086c2ff

  Copy full SHA for 086c2ff

• example script

  

  Sara Adkins

  committed
  64096c7

  Copy full SHA for 64096c7

Commits on Apr 2, 2024

• POC SFT sparse trainer

  

  Sara Adkins

  committed
  e243b46

  Copy full SHA for e243b46

• use sft data functionality

  

  Sara Adkins

  committed
  c84cd60

  Copy full SHA for c84cd60

• update unit tests

  

  Sara Adkins

  committed
  4f938cb

  Copy full SHA for 4f938cb

• move examples folder

  

  Sara Adkins

  committed
  3821606

  Copy full SHA for 3821606

• clarity comments

  

  Sara Adkins

  committed
  4f619bd

  Copy full SHA for 4f619bd

• barest bones trainer

  

  Sara Adkins

  committed
  e425523

  Copy full SHA for e425523

Commits on Apr 3, 2024

• style

  

  Sara Adkins

  committed
  3c9a2e3

  Copy full SHA for 3c9a2e3

• tweaks to work with distillation and FSDP

  

  Sara Adkins

  committed
  2ed1bb2

  Copy full SHA for 2ed1bb2

• Merge branch 'main' into sa/sft_trainer_mixin

  

  Sara Adkins

  committed
  118c7c1

  Copy full SHA for 118c7c1

Commits on Apr 9, 2024

• tests and readme

  

  Sara Adkins

  committed
  1f61081

  Copy full SHA for 1f61081

• naming

  

  Sara Adkins

  committed
  44cb166

  Copy full SHA for 44cb166

• Merge branch 'main' into sa/sft_trainer_mixin

  

  Sara Adkins

  authored
  db5060c

  Copy full SHA for db5060c

Commits on Apr 10, 2024

• Merge branch 'main' into sa/sft_trainer_mixin

  

  Sara Adkins

  authored
  e8021e6

  Copy full SHA for e8021e6

• Merge branch 'main' into sa/sft_trainer_mixin

  

  Sara Adkins

  authored
  7648a7f

  Copy full SHA for 7648a7f

Commits on Apr 11, 2024

• Merge branch 'main' into sa/sft_trainer_mixin

  

  Sara Adkins

  authored
  02aec71

  Copy full SHA for 02aec71

Commits on Apr 16, 2024

• move examples

  

  Sara Adkins

  committed
  0051eaf

  Copy full SHA for 0051eaf

• quality

  

  Sara Adkins

  committed
  7630dd7

  Copy full SHA for 7630dd7

Commits on Apr 22, 2024

• Merge branch 'main' into sa/sft_trainer_mixin

  

  Sara Adkins

  authored
  7b289d2

  Copy full SHA for 7b289d2