ovshake commited on
Commit
b112dcf
1 Parent(s): 6724ca0

change requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -4
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  absl-py==1.2.0
2
  accelerate==0.12.0
3
  aiohttp==3.8.1
@@ -75,9 +76,9 @@ tensorboard-plugin-wit==1.8.1
75
  tensorboardX==2.5.1
76
  tokenizers==0.13.2
77
  toml==0.10.2
78
- torch==1.13.0+cu116
79
- torchaudio==0.13.0+cu116
80
- torchvision==0.14.0+cu116
81
  tqdm==4.64.0
82
  traitlets==5.3.0
83
  transformers==4.24.0
@@ -88,4 +89,5 @@ Werkzeug==2.2.2
88
  xxhash==3.0.0
89
  yarl==1.8.1
90
  zipp==3.8.1
91
- rembg
 
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu116
2
  absl-py==1.2.0
3
  accelerate==0.12.0
4
  aiohttp==3.8.1
 
76
  tensorboardX==2.5.1
77
  tokenizers==0.13.2
78
  toml==0.10.2
79
+ torch
80
+ torchaudio
81
+ torchvision
82
  tqdm==4.64.0
83
  traitlets==5.3.0
84
  transformers==4.24.0
 
89
  xxhash==3.0.0
90
  yarl==1.8.1
91
  zipp==3.8.1
92
+ rembg
93
+ huggingface-hub