-
Notifications
You must be signed in to change notification settings - Fork 7
/
env.yml
47 lines (42 loc) · 724 Bytes
/
env.yml
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
channels:
- conda-forge
dependencies:
- python >=3.9
- pip
- tqdm
- loguru
- typer
- universal_pathlib
# Scientific
- datamol
- pandas <=2.1.1
- numpy
- pytorch >=2.0
- transformers
- datasets
- tokenizers
- accelerate >=0.28.0 # for accelerator_config update
- evaluate
- wandb
- huggingface_hub
# Optional
- deepspeed
# dev
- black >=24
- ruff
- pytest >=6.0
- nbconvert
- jupyterlab
- nbconvert
- ipywidgets
- pip:
- mkdocs <1.6.0
- mkdocs-material >=7.1.1
- mkdocs-material-extensions
- mkdocstrings
- mkdocstrings-python
- mkdocs-jupyter
- markdown-include
- markdown-include
- mdx_truly_sane_lists
- mike >=1.0.0