add detectron2

pull/139/head
jwyang 2023-10-05 07:55:19 +00:00
parent 40dbcbc358
commit b120d51969
1 changed files with 7 additions and 1 deletions

View File

@ -12,6 +12,7 @@ classifiers = [
"Programming Language :: Python :: 3",
"License :: OSI Approved :: Apache Software License",
]
dependencies = [
"torch==2.1.0",
"torchvision==0.16.0",
@ -46,9 +47,14 @@ dependencies = [
"deepspeed==0.10.3",
"wandb==0.15.12",
"infinibatch==0.1.1",
"gradio==3.42.0"
"gradio==3.42.0",
"openai-whisper",
]
[tool.poetry.dependencies]
detectron2 = {git = "https://github.com/MaureenZOU/detectron2-xyz.git"}
[project.urls]
"Paper" = "https://arxiv.org/abs/2304.06718"
"Code" = "https://github.com/UX-Decoder/Segment-Everything-Everywhere-All-At-Once/tree/v1.0"