From 2568d94e50eaa666c6eb38779ac53944e28e7523 Mon Sep 17 00:00:00 2001 From: Zach Nussbaum 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 00000000..6db10a88 --- /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 00000000..8a91fd74 --- /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