diff --git a/inference/requirements.txt b/inference/requirements.txt index c82cd4a..76c961f 100644 --- a/inference/requirements.txt +++ b/inference/requirements.txt @@ -1,4 +1,4 @@ torch==2.6.0 -triton==3.0.0 +git+https://github.com/NripeshN/triton.git@main#subdirectory=python transformers==4.46.3 safetensors==0.4.5 \ No newline at end of file