Update requirements.txt
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
|
@@ -5,6 +5,6 @@ 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 |
|
| 8 |
-
torchaudio==2.0.1+
|
| 9 |
-f https://download.pytorch.org/whl/cpu
|
| 10 |
|
|
|
|
| 5 |
plotly>=5.3.0 # Required for data visualization
|
| 6 |
torch>=1.9.0 # Required for PyTorch (for GPT-2 model)
|
| 7 |
|
| 8 |
+
torchaudio==2.0.1+gpu # Replace with the corresponding version
|
| 9 |
-f https://download.pytorch.org/whl/cpu
|
| 10 |
|