feat: env for conda, pip

2024-10-01 01:06:10 -04: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