Skip to content

Commit b864962

Browse files
authored
Add dependency for ninja (#244)
This pr adds dependency for ninja for torch extension
1 parent 47ea96c commit b864962

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

pyproject.toml

+1
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ dependencies = [
2020
"torch>=1.10.0",
2121
"transformers>=4.38.0",
2222
"triton; platform_system == 'linux' and platform_machine == 'x86_64'",
23+
"ninja",
2324
]
2425
dynamic = ["version"]
2526

0 commit comments

Comments
 (0)