Update requirements.txt

This commit is contained in:
Jianan Xing 2024-08-19 16:49:40 -07:00 committed by GitHub
parent 49ed4adff5
commit de8eb63f09
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -4,4 +4,4 @@ torch
tqdm
more-itertools
tiktoken
triton>=2.0.0,<3;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2"
triton>=2.0.0;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2"