Fix triton env marker (#1887)

2026-01-09 21:38:09 -05:00 · 2023-12-11 23:39:08 +08:00
parent e58f288045
commit 8bc8860694
2 changed files with 4 additions and 4 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,3 +4,4 @@ torch
 tqdm
 more-itertools
 tiktoken
+triton>=2.0.0,<3;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2"