[FIX] torchvision requirement(yaml) #59

prs-eth · Apr 28, 2024 · a6dd76a · a6dd76a
1 parent 354c01d
commit a6dd76a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/environment.yaml b/environment.yaml
@@ -11,5 +11,6 @@ dependencies:
   - python=3.10.12
   - pytorch=2.0.1
   - pytorch-cuda=11.7
+  - torchvision=0.15.2
   - pip:
     - transformers==4.32.1