20 lines
469 B
TOML
20 lines
469 B
TOML
[project]
|
|
name = "prain"
|
|
version = "0.1.0"
|
|
description = "Per-expert VRAM/RAM/disk placement for MoE models served by vLLM"
|
|
requires-python = ">=3.11"
|
|
dependencies = [] # torch and vllm come from the gfx1030 vLLM image
|
|
|
|
[project.scripts]
|
|
prain = "prain.cli:main"
|
|
|
|
[project.entry-points."vllm.general_plugins"]
|
|
prain = "prain.plugin:register"
|
|
|
|
[build-system]
|
|
requires = ["setuptools>=64"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[tool.setuptools]
|
|
packages = ["prain"]
|