train new models based on stablelm + properly add new response types

2026-01-08 21:28:05 -05:00 · 2024-02-13 20:21:51 -05:00
parent 1da51c9fb6
commit 411276408b
26 changed files with 655 additions and 62 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,3 +5,4 @@ peft
 bitsandbytes
 webcolors
 pandas
+# flash-attn