diff --git a/inference/requirements.txt b/inference/requirements.txt index 68a60d0..700cf1d 100644 --- a/inference/requirements.txt +++ b/inference/requirements.txt @@ -1,4 +1,4 @@ -torch==2.4.1 -triton==3.0.0 +torch==2.6.0 +triton==3.2.0 transformers==4.46.3 -safetensors==0.4.5 \ No newline at end of file +safetensors==0.4.5