feat: env for conda, pip

ritualage · Mar 25, 2023 · 2568d94 · 2568d94
1 parent e8d5ad9
commit 2568d94
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/env.yaml 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
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,12 @@
+accelerate
+datasets
+torchmetrics
+evaluate
+transformers
+wandb
+pip
+peft
+nodelist-inflator
+deepspeed
+sentencepiece
+jsonlines