From b120d519695a9090c273b07fbe0ac57260daba16 Mon Sep 17 00:00:00 2001 From: jwyang Date: Thu, 5 Oct 2023 07:55:19 +0000 Subject: [PATCH] add detectron2 --- pyproject.toml | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 8071ea9..92ff46c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -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"