peregrine/scripts/classifier_service/environment.yml

20 lines
366 B
YAML

name: job-seeker-classifiers
channels:
- pytorch
- nvidia
- conda-forge
- defaults
dependencies:
- python=3.11
- pip
- pip:
- torch>=2.1.0
- transformers>=4.40.0
- accelerate>=0.26.0
- sentencepiece>=0.1.99
- protobuf>=4.25.0
- gliclass>=0.1.0
- FlagEmbedding>=1.2.0
- pyyaml>=6.0
- tqdm>=4.66.0
- pytest>=8.0.0