From 2568d94e50eaa666c6eb38779ac53944e28e7523 Mon Sep 17 00:00:00 2001
From: Zach Nussbaum <zanussbaum@gmail.com>
Date: Sat, 25 Mar 2023 16:16:40 +0000
Subject: [PATCH] feat: env for conda, pip

---
 env.yaml         | 20 ++++++++++++++++++++
 requirements.txt | 12 ++++++++++++
 2 files changed, 32 insertions(+)
 create mode 100644 env.yaml
 create mode 100644 requirements.txt

diff --git a/env.yaml b/env.yaml
new file mode 100644
index 000000000000..6db10a88dab6
--- /dev/null
+++ b/env.yaml
@@ -0,0 +1,20 @@
+name: vicuna
+channels:
+  - conda-forge
+  - pytorch
+  - nvidia
+  - huggingface
+dependencies:
+  - python=3.8
+  - accelerate
+  - datasets
+  - torchmetrics
+  - evaluate
+  - transformers
+  - wandb
+  - jsonlines
+  - pip:
+    - peft
+    - nodelist-inflator
+    - deepspeed
+    - sentencepiece
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 000000000000..8a91fd74b60d
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,12 @@
+accelerate
+datasets
+torchmetrics
+evaluate
+transformers
+wandb
+pip
+peft
+nodelist-inflator
+deepspeed
+sentencepiece
+jsonlines
\ No newline at end of file