gpt4all/gpt4all-training/env.yaml

20 lines
285 B
YAML
Raw Normal View History

2023-03-25 16:16:40 +00:00
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