override the entire create_optimzier method

2024-03-19 23:19:56 -04:00
parent a236f5eab5
commit e6b78c1fca
2 changed files with 63 additions and 15 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -43,3 +43,4 @@ trl @ git+https://github.com/huggingface/trl.git@304e208f778a5442c30cdda50034822
 fastcore>=1.5.29

 lpmm @ git+https://github.com/thu-ml/low-bit-optimizers.git@main
+yacs