Pin triton==2.2.0

stanford-crfm · Jan 6, 2025 · 40c4a6e · 40c4a6e
1 parent 088b3fd
commit 40c4a6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -297,7 +297,7 @@ transformers==4.45.2
 transformers-stream-generator==0.0.5
 trio==0.28.0
 trio-websocket==0.11.1
-triton==3.1.0
+triton==2.2.0
 typer==0.15.1
 types-requests==2.31.0.6
 types-urllib3==1.26.25.14