engr-awaisjamal commited on
Commit
ae8d541
·
verified ·
1 Parent(s): 6676a2f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -0
requirements.txt CHANGED
@@ -4,3 +4,8 @@ pandas>=1.3.0 # Required for data manipulation
4
  transformers>=4.30.0 # For Hugging Face GPT-2 model
5
  plotly>=5.3.0 # Required for data visualization
6
  torch>=1.9.0 # Required for PyTorch (for GPT-2 model)
 
 
 
 
 
 
4
  transformers>=4.30.0 # For Hugging Face GPT-2 model
5
  plotly>=5.3.0 # Required for data visualization
6
  torch>=1.9.0 # Required for PyTorch (for GPT-2 model)
7
+ torch==2.0.1+cpu # Replace with the desired version
8
+ torchvision==0.15.2+cpu # Replace with the corresponding version
9
+ torchaudio==2.0.1+cpu # Replace with the corresponding version
10
+ -f https://download.pytorch.org/whl/cpu
11
+