File size: 477 Bytes
ce00289 0609e6d ce00289 |
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 |
name: llmtt
channels:
- pytorch
- nvidia
- conda-forge
dependencies:
- python==3.12
- pytorch
- pytorch-cuda=11.8
- nodejs
- yarn
- pip
- pip:
- datasets
- einops
- fancy_einsum
- jaxtyping
- networkx
- plotly
- pyinstrument
- setuptools
- streamlit
- streamlit_extras
- tokenizers
- transformer_lens
- transformers
- pytest # fixes wrong dependencies of transformer_lens
|