feat: add job-seeker-classifiers conda env for HF classifier benchmark

2026-02-26 23:43:41 -08:00 · 2026-02-26 23:43:41 -08:00 · 99f0f5b277
commit 99f0f5b277
parent cf98044f6f
1 changed files with 20 additions and 0 deletions
--- a/scripts/classifier_service/environment.yml
+++ b/scripts/classifier_service/environment.yml
@ -0,0 +1,20 @@
+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