feat: env for conda, pip

2024-09-19 15:25:53 +00:00 · 2023-03-25 16:16:40 +00:00 · 2023-03-25 16:16:40 +00:00 · 2568d94e50
commit 2568d94e50
parent e8d5ad9fc2
2 changed files with 32 additions and 0 deletions
--- 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
--- a/requirements.txt
+++ b/requirements.txt
@ -0,0 +1,12 @@
+accelerate
+datasets
+torchmetrics
+evaluate
+transformers
+wandb
+pip
+peft
+nodelist-inflator
+deepspeed
+sentencepiece
+jsonlines