| 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 | |