Update requirements.txt

requirements.txt

@@ -17,6 +17,9 @@ exceptiongroup==1.2.2
 fastapi==0.115.11  
 ffmpy==0.5.0  
 filelock==3.17.0  
+torch==2.1.2
+torchaudio==2.1.2
+torchvision==0.16.2
 flash-attn==2.7.1.post4  
 frozenlist==1.5.0  
 fsspec==2024.12.0  
@@ -76,9 +79,6 @@ sympy==1.13.3
 timm==1.0.14  
 tokenizers==0.20.3  
 tomlkit==0.13.2  
-torch==2.1.2+cu118  
-torchaudio==2.1.2
-torchvision==0.16.2
 tqdm==4.67.1  
 transformers==4.45.0  
 triton==2.1.0