PrompRL / environment.yml
wangfuyun's picture
Add UniRL inference code
522bf24 verified
name: unirl
channels:
- defaults
dependencies:
- _libgcc_mutex=0.1=main
- _openmp_mutex=5.1=1_gnu
- bzip2=1.0.8=h5eee18b_6
- ca-certificates=2025.2.25=h06a4308_0
- expat=2.7.1=h6a678d5_0
- ld_impl_linux-64=2.40=h12ee557_0
- libffi=3.4.4=h6a678d5_1
- libgcc-ng=11.2.0=h1234567_1
- libgomp=11.2.0=h1234567_1
- libstdcxx-ng=11.2.0=h1234567_1
- libuuid=1.41.5=h5eee18b_0
- libxcb=1.17.0=h9b100fa_0
- ncurses=6.4=h6a678d5_0
- openssl=3.0.16=h5eee18b_0
- pip=25.1=pyhc872135_2
- pthread-stubs=0.3=h0ce48e5_1
- python=3.11.13=h1a3bd86_0
- readline=8.2=h5eee18b_0
- setuptools=78.1.1=py311h06a4308_0
- sqlite=3.45.3=h5eee18b_0
- tk=8.6.14=h993c535_1
- wheel=0.45.1=py311h06a4308_0
- xorg-libx11=1.8.12=h9b100fa_1
- xorg-libxau=1.0.12=h9b100fa_0
- xorg-libxdmcp=1.1.5=h9b100fa_0
- xorg-xorgproto=2024.1=h5eee18b_1
- xz=5.6.4=h5eee18b_1
- zlib=1.2.13=h5eee18b_1
- pip:
- accelerate==1.7.0
- aiohappyeyeballs==2.6.1
- aiohttp==3.12.9
- aiosignal==1.3.2
- airportsdata==20250523
- annotated-types==0.7.0
- anthropic==0.54.0
- antlr4-python3-runtime==4.13.2
- anyio==4.9.0
- astor==0.8.1
- asttokens==3.0.0
- attrs==25.3.0
- av==14.4.0
- bitsandbytes==0.46.0
- blake3==1.0.5
- cachetools==6.0.0
- certifi==2025.4.26
- charset-normalizer==3.4.2
- click==8.2.1
# - clip==1.0
- cloudpickle==3.1.1
- compressed-tensors==0.9.4
- contourpy==1.3.2
- cupy-cuda12x==13.4.1
- cycler==0.12.1
- datasets==3.6.0
- decorator==5.2.1
- deepspeed==0.15.4
- depyf==0.18.0
# - diffusers==0.34.0.dev0
- dill==0.3.8
- diskcache==5.6.3
- distro==1.9.0
- dnspython==2.7.0
- docker-pycreds==0.4.0
- einops==0.8.1
- email-validator==2.2.0
- executing==2.2.0
- fastapi==0.115.12
- fastapi-cli==0.0.7
- fastrlock==0.8.3
- filelock==3.18.0
# - flash-attn==2.7.4.post1
- fonttools==4.58.4
- frozenlist==1.6.2
- fsspec==2025.3.0
- ftfy==6.3.1
- gguf==0.17.0
- gitdb==4.0.12
- gitpython==3.1.44
- googleapis-common-protos==1.70.0
- grpcio==1.72.1
- h11==0.16.0
- hf-transfer==0.1.9
- hf-xet==1.1.3
- hjson==3.1.0
- httpcore==1.0.9
- httptools==0.6.4
- httpx==0.28.1
- huggingface-hub==0.32.4
- idna==3.10
- importlib-metadata==8.7.0
- inquirerpy==0.3.4
- interegular==0.3.3
- ipython==9.3.0
- ipython-pygments-lexers==1.1.1
- jedi==0.19.2
- jinja2==3.1.6
- jiter==0.10.0
- jsonschema==4.24.0
- jsonschema-specifications==2025.4.1
- kiwisolver==1.4.8
- lark==1.2.2
- latex2sympy2-extended==1.10.1
- liger-kernel==0.5.2
- llguidance==0.7.29
- llvmlite==0.44.0
- lm-format-enforcer==0.10.11
- markdown-it-py==3.0.0
- markupsafe==3.0.2
- math-verify==0.7.0
- matplotlib==3.10.3
- matplotlib-inline==0.1.7
- mdurl==0.1.2
- mistral-common==1.5.6
- mpmath==1.3.0
- msgpack==1.1.0
- msgspec==0.19.0
- multidict==6.4.4
- multiprocess==0.70.16
- nest-asyncio==1.6.0
- networkx==3.5
- ninja==1.11.1.4
- numba==0.61.2
- numpy==2.2.6
- nvidia-cublas-cu12==12.6.4.1
- nvidia-cuda-cupti-cu12==12.6.80
- nvidia-cuda-nvrtc-cu12==12.6.77
- nvidia-cuda-runtime-cu12==12.6.77
- nvidia-cudnn-cu12==9.5.1.17
- nvidia-cufft-cu12==11.3.0.4
- nvidia-cufile-cu12==1.11.1.6
- nvidia-curand-cu12==10.3.7.77
- nvidia-cusolver-cu12==11.7.1.2
- nvidia-cusparse-cu12==12.5.4.2
- nvidia-cusparselt-cu12==0.6.3
- nvidia-nccl-cu12==2.26.2
- nvidia-nvjitlink-cu12==12.6.85
- nvidia-nvtx-cu12==12.6.77
- openai==1.84.0
- opencv-python-headless==4.11.0.86
- opentelemetry-api==1.34.0
- opentelemetry-exporter-otlp==1.34.0
- opentelemetry-exporter-otlp-proto-common==1.34.0
- opentelemetry-exporter-otlp-proto-grpc==1.34.0
- opentelemetry-exporter-otlp-proto-http==1.34.0
- opentelemetry-proto==1.34.0
- opentelemetry-sdk==1.34.0
- opentelemetry-semantic-conventions==0.55b0
- opentelemetry-semantic-conventions-ai==0.4.9
- outlines==0.1.11
- outlines-core==0.1.26
- packaging==25.0
- pandas==2.3.0
- parso==0.8.4
- partial-json-parser==0.2.1.1.post5
- peft==0.17.1
- pexpect==4.9.0
- pfzy==0.3.4
- pillow==11.2.1
- platformdirs==4.3.8
- prometheus-client==0.22.1
- prometheus-fastapi-instrumentator==7.1.0
- prompt-toolkit==3.0.51
- propcache==0.3.1
- protobuf==5.29.5
- psutil==7.0.0
- ptyprocess==0.7.0
- pure-eval==0.2.3
- py-cpuinfo==9.0.0
- pyarrow==20.0.0
- pycountry==24.6.1
- pydantic==2.11.5
- pydantic-core==2.33.2
- pygments==2.19.1
- pyparsing==3.2.3
- python-dateutil==2.9.0.post0
- python-dotenv==1.1.0
- python-json-logger==3.3.0
- python-multipart==0.0.20
- pytz==2025.2
- pyyaml==6.0.2
- pyzmq==26.4.0
- qwen-vl-utils==0.0.11
- ray==2.46.0
- referencing==0.36.2
- regex==2024.11.6
- requests==2.32.3
- rich==14.0.0
- rich-toolkit==0.14.7
- rpds-py==0.25.1
- safetensors==0.5.3
- scipy==1.15.3
- seaborn==0.13.2
- sentencepiece==0.2.0
- sentry-sdk==2.29.1
- setproctitle==1.3.6
- shellingham==1.5.4
- six==1.17.0
- smmap==5.0.2
- sniffio==1.3.1
- stack-data==0.6.3
- starlette==0.46.2
- sympy==1.14.0
- tabulate==0.9.0
- tiktoken==0.9.0
- timm==0.6.13
- tokenizers==0.21.1
- torch==2.7.0
- torchaudio==2.7.0
- torchvision==0.22.0
- tqdm==4.67.1
- traitlets==5.14.3
- transformers==4.51.3
- triton==3.3.0
- trl==0.19.0
- typer==0.16.0
- typing-extensions==4.14.0
- typing-inspection==0.4.1
- tzdata==2025.2
- urllib3==2.4.0
- utils==1.0.2
- uvicorn==0.34.3
- uvloop==0.21.0
- vllm==0.9.0.1
- wandb==0.18.3
- watchfiles==1.0.5
- wcwidth==0.2.13
- websockets==15.0.1
- xformers==0.0.30
- xgrammar==0.1.19
- xxhash==3.5.0
- yarl==1.20.0
- zipp==3.22.0
- tensorboardX==2.6.4