ROLL/requirements_em_local_debug.txt at main · alibaba/ROLL · GitHub

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
torch
torchvision
torchaudio

# mac
py-cpuinfo
ray<=2.46.0,>=2.40.0
numpy<2.0a0,>=1.25
tensordict
sympy==1.13.1
transformers==4.51.2
modelscope
datasets==3.1.0
tqdm
peft==0.12.0
tyro>=0.5.7
accelerate==0.34.2
pydantic
pytest
loralib
einops
isort
jsonlines
deprecated
trl==0.9.6
pyext
dacite
codetiming
more_itertools
pybase64
matplotlib

wandb

math-verify
openai

gym
gymnasium[toy-text]
gym_sokoban

hydra-core
omegaconf
latex2sympy2==1.5.4
latex2sympy2_extended==1.10.1
antlr4-python3-runtime==4.9.3

./mcore_adapter

-r requirements_vision.txt