config chooser, update readme instructions, device config, llama flash attention, debug out the labels, fix config key checks, other bugfixes

2023-04-14 12:18:56 -04:00
parent a6028d302e
commit f2a2029d0d
7 changed files with 283 additions and 9 deletions
--- a/setup.cfg
+++ b/setup.cfg
@@ -17,6 +17,12 @@ install_requires =
    fire
    PyYAML == 6.0
    black
+    bitsandbytes
+    datasets
+    accelerate
+    sentencepiece
+    wandb
+    flash-attn

 [options.packages.find]
 where = src