ovshake
commited on
Commit
•
b112dcf
1
Parent(s):
6724ca0
change requirements
Browse files- 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
|
| 79 |
-
torchaudio
|
| 80 |
-
torchvision
|
| 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
|