From a6dd76a9eaa9084863e906cb02688b07981d37a6 Mon Sep 17 00:00:00 2001
From: Bingxin Ke <bingke@ethz.ch>
Date: Sun, 28 Apr 2024 15:41:53 +0200
Subject: [PATCH] [FIX] torchvision requirement(yaml) #59

---
 environment.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/environment.yaml b/environment.yaml
index 4f2ad2b..77920f7 100644
--- 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