Vara1605454 commited on
Commit
eaff076
·
verified ·
1 Parent(s): b79adb2

Updated file

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,7 +1,8 @@
1
 
2
  # Core PyTorch (use CPU-only versions for faster install)
3
- torch==2.0.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
4
- torchvision==0.15.2+cpu -f https://download.pytorch.org/whl/torch_stable.html
 
5
 
6
  # Image processing
7
  Pillow==9.5.0
 
1
 
2
  # Core PyTorch (use CPU-only versions for faster install)
3
+ # Core PyTorch (standard versions - will auto-detect CPU/GPU)
4
+ torch==2.0.1
5
+ torchvision==0.15.2
6
 
7
  # Image processing
8
  Pillow==9.5.0