From 8ae5ff20093f74671bcdf2c9d4e078839c7a7732 Mon Sep 17 00:00:00 2001
From: liuzhenwei <zhenweiliu@habana.ai>
Date: Thu, 23 Jan 2025 16:35:46 +0800
Subject: [PATCH] [Hardware][Gaudi][BugFix] Fix dataclass error due to triton
 package update (#12338)

Signed-off-by: zhenwei <zhenweiliu@habana.ai>
---
 requirements-hpu.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements-hpu.txt b/requirements-hpu.txt
index f4fb89ef..63a5f8b1 100644
--- a/requirements-hpu.txt
+++ b/requirements-hpu.txt
@@ -3,7 +3,7 @@
 
 # Dependencies for HPU code
 ray
-triton
+triton==3.1.0
 pandas
 tabulate
 setuptools>=61