From bec8aedeabbc1d67daa2c1d9622112deb3eb4163 Mon Sep 17 00:00:00 2001
From: onadegibert <odegiber@gmail.com>
Date: Fri, 5 Jul 2024 13:11:56 +0300
Subject: [PATCH] added pytorch-rocm dependency

---
 envs/hf.yml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/envs/hf.yml b/envs/hf.yml
index bad268c00..10aab180b 100644
--- a/envs/hf.yml
+++ b/envs/hf.yml
@@ -11,4 +11,6 @@ dependencies:
     - transformers==4.38.2
     - sentencepiece==0.1.99
     - accelerate==0.31.0
-    - torch==2.2.1
\ No newline at end of file
+    - torch==2.2.1+rocm5.3
+    - torchvision==0.15.2+rocm5.3
+    - torchaudio==0.13.0+rocm5.3
\ No newline at end of file