Update pyproject.toml
Browse files- pyproject.toml +5 -6
pyproject.toml
CHANGED
@@ -13,15 +13,14 @@ classifiers = [
|
|
13 |
"License :: OSI Approved :: Apache Software License",
|
14 |
]
|
15 |
dependencies = [
|
16 |
-
"torch
|
17 |
-
"transformers
|
18 |
-
"einops", "fastapi", "markdown2[all]", "numpy
|
19 |
"requests", "sentencepiece", "tokenizers>=0.12.1",
|
20 |
"uvicorn", "tensorboard", "open_clip_torch",
|
21 |
-
"shortuuid", "httpx
|
22 |
-
"deepspeed
|
23 |
"peft==0.4.0",
|
24 |
-
"transformers==4.31.0",
|
25 |
"accelerate==0.21.0",
|
26 |
"bitsandbytes==0.41.0",
|
27 |
"scikit-learn==1.2.2",
|
|
|
13 |
"License :: OSI Approved :: Apache Software License",
|
14 |
]
|
15 |
dependencies = [
|
16 |
+
"torch", "torchvision",
|
17 |
+
"transformers",
|
18 |
+
"einops", "fastapi", "markdown2[all]", "numpy",
|
19 |
"requests", "sentencepiece", "tokenizers>=0.12.1",
|
20 |
"uvicorn", "tensorboard", "open_clip_torch",
|
21 |
+
"shortuuid", "httpx",
|
22 |
+
"deepspeed",
|
23 |
"peft==0.4.0",
|
|
|
24 |
"accelerate==0.21.0",
|
25 |
"bitsandbytes==0.41.0",
|
26 |
"scikit-learn==1.2.2",
|