yuanwenyue commited on
Commit
dd7b0b0
1 Parent(s): 55a66b7

Create requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +13 -0
requirements.txt ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu117
2
+ torch==2.0.0
3
+ torchvision==0.15.0
4
+ torch_kmeans
5
+ numpy==1.26.4
6
+ imageio
7
+ gradio
8
+ timm==0.9.10
9
+ tqdm
10
+ scikit-learn
11
+ pillow
12
+ matplotlib
13
+ albumentations==1.3.1