ivmpfa commited on
Commit
39eff94
Β·
verified Β·
1 Parent(s): 11838c6

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  transformers==4.33.0
2
- torch==1.13.1+cpu # Stable CPU version for GPT-2
3
- numpy==1.24.4 # Avoid NumPy 2.x conflicts
 
4
  gradio==3.36.3
 
1
  transformers==4.33.0
2
+ torch==1.13.1
3
+ torchvision==0.14.1 # Matches PyTorch 1.13.1
4
+ numpy==1.24.4
5
  gradio==3.36.3