Skip to content
Merged
Show file tree
Hide file tree
Changes from 5 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -28,3 +28,5 @@ venv/

# Tox envs
.tox

.python-version
2 changes: 1 addition & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ ARG BASE_UBI_IMAGE_TAG=9.5
ARG PYTHON_VERSION=3.12

### Build layer
FROM quay.io/vllm/vllm:0.8.5.0_cu128 as build
FROM quay.io/modh/vllm:rhoai-2.23-cuda as build

ARG PYTHON_VERSION
ENV PYTHON_VERSION=${PYTHON_VERSION}
Expand Down
29 changes: 13 additions & 16 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -2,28 +2,22 @@
name = "vllm-detector-adapter"
version = "0.7.1"
authors = [
{ name="Gaurav Kumbhat", email="[email protected]" },
{ name="Evaline Ju", email="[email protected]" },
{ name = "Gaurav Kumbhat", email = "[email protected]" },
{ name = "Evaline Ju", email = "[email protected]" },
]
description = "A lightweight adapter layer that provides detector APIs on top of vllm"
readme = "README.md"
requires-python = ">=3.11"
classifiers = [
"Programming Language :: Python :: 3"
]
classifiers = ["Programming Language :: Python :: 3"]

dependencies = [
"orjson>=3.10.16,<3.11",
]
dependencies = ["orjson>=3.10.16,<3.11"]

[project.optional-dependencies]
vllm-tgis-adapter = [
"vllm-tgis-adapter>=0.7.0,<0.7.2"
]
vllm-tgis-adapter = ["vllm-tgis-adapter>=0.7.0,<0.7.2"]
vllm = [
# Note: 0.8.4 has a triton bug on Mac
"vllm @ git+https://github.com/vllm-project/vllm.git@v0.8.5 ; sys_platform == 'darwin'",
"vllm>=0.7.2,<0.8.6 ; sys_platform != 'darwin'",
"vllm @ git+https://github.com/vllm-project/vllm.git@v0.9.0 ; sys_platform == 'darwin'",
"vllm>=0.7.2,<0.9.1 ; sys_platform != 'darwin'",
]

## Dev Extra Sets ##
Expand All @@ -38,10 +32,13 @@ dev-test = [

dev-fmt = [
"ruff==0.4.7",
"pre-commit>=3.0.4,<4.0",
"pydeps>=1.12.12,<2",
"pre-commit>=3.0.4,<4.0",
"pydeps>=1.12.12,<2"
]

[tool.setuptools.packages.find]
where = [""]
include = ["vllm_detector_adapter", "vllm_detector_adapter*"]
include = [
"vllm_detector_adapter",
"vllm_detector_adapter*"
]