Skip to content

GitLab

Unverified Commit 8849043a authored by Casper's avatar Casper Committed by GitHub
Browse files

Fix triton dependency (#350)

parent 969b290e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
setup.py
View file @ 8849043a
...@@ -89,6 +89,7 @@ requirements = [ ...@@ -89,6 +89,7 @@ requirements = [
"transformers>=4.35.0", "transformers>=4.35.0",
"tokenizers>=0.12.1", "tokenizers>=0.12.1",
"typing_extensions>=4.8.0", "typing_extensions>=4.8.0",
"triton",
"accelerate", "accelerate",
"datasets", "datasets",
"zstandard", "zstandard",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment