File size: 754 Bytes
6bb0065 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | # environment.yml
name: allm # The name of the environment
channels: # The conda channels to search for packages
# - pytorch
- conda-forge
# - dnachun
# - anaconda
channel_priority: strict
dependencies:
# Packages to install with conda
- python=3.11.3
#- pytorch-cuda=12.4
#- pytorch
# - numpy
- transformers>=4.55
- einops
- jaxtyping
- tensorboard
- omegaconf
- accelerate
- peft
- wandb
- scipy
- pandas
- matplotlib
- scikit-image
- scikit-learn
- joblib
- pillow
- datasets
## NO - huggingface_hub
- tqdm
- nltk
- future
- defusedxml
- ipdb
- torchinfo
- timm
- graphviz #anaconda::graphviz
- dnachun::torchviz
- pip:
# - draccus
- fraction
- vllm
|