From 022dd7d1bc49c81a93154a38ec048e72f99c8122 Mon Sep 17 00:00:00 2001
From: djliden <7102904+djliden@users.noreply.github.com>
Date: Wed, 10 Apr 2024 15:51:31 -0500
Subject: [PATCH] adds qlora nb requirements

---
 notebooks/drafts/olmo_peft_requirements.txt | 7 +++++++
 1 file changed, 7 insertions(+)
 create mode 100644 notebooks/drafts/olmo_peft_requirements.txt

diff --git a/notebooks/drafts/olmo_peft_requirements.txt b/notebooks/drafts/olmo_peft_requirements.txt
new file mode 100644
index 0000000..a8c42ff
--- /dev/null
+++ b/notebooks/drafts/olmo_peft_requirements.txt
@@ -0,0 +1,7 @@
+transformers==4.37.2
+accelerate
+peft
+ai2-olmo
+bitsandbytes
+mlflow
+pynvml
\ No newline at end of file