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/requirements.txt
+++ b/requirements.txt
@@ -4,3 +4,9 @@ attrdict
 fire
 PyYAML==6.0
 black
+bitsandbytes
+datasets
+accelerate
+sentencepiece
+wandb
+flash-attn