mahnerak's picture
Initial Commit πŸš€
ce00289
raw
history blame
471 Bytes
name: llmtt
channels:
- pytorch
- nvidia
- conda-forge
dependencies:
- python
- 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