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