yslan commited on
Commit
4907461
1 Parent(s): 6eb678b
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
- torch
 
2
  xformers==0.0.26.post1
3
- torchvision
4
  absl-py
5
  tensorboard
6
  lpips
 
1
+ torch==2.3.0+cu118
2
+ torchvision==0.18.0+cu118
3
  xformers==0.0.26.post1
 
4
  absl-py
5
  tensorboard
6
  lpips