mirror/OpenLLM
1
0
Fork 0
mirror of https://github.com/bentoml/OpenLLM.git synced 2026-06-14 03:20:18 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(dependencies): lock build < 1 for now (#643)

Browse Source 
Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
This commit is contained in:
Aaron Pham
2023-11-14 00:36:08 -05:00
committed by GitHub
parent b30a412398
commit 0bf6ec7537
2 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
openllm-python/pyproject.toml
View File

@@ -46,6 +46,7 @@ dependencies = [
"optimum>=1.12.0",
"accelerate",
"ghapi",
"build[virtualenv]<1",
"click>=8.1.3",
"cuda-python;platform_system!=\"Darwin\"",
"bitsandbytes<0.42",

7
tools/dependencies.py
View File

@@ -7,20 +7,16 @@ import typing as t
import inflection
import tomlkit
from ghapi.all import GhApi
if t.TYPE_CHECKING:
from tomlkit.items import Array
from tomlkit.items import Table
from tomlkit.items import Array, Table
ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, os.path.join(ROOT, 'openllm-python', 'src'))
import openllm
_OWNER, _REPO = 'bentoml', 'openllm'
@@ -157,6 +153,7 @@ _BASE_DEPENDENCIES = [
Dependencies(name='optimum', lower_constraint='1.12.0'),
Dependencies(name='accelerate'),
Dependencies(name='ghapi'),
Dependencies(name='build', upper_constraint='1', extensions=['virtualenv']),
Dependencies(name='click', lower_constraint='8.1.3'),
Dependencies(name='cuda-python', platform=('Darwin', 'ne')),
Dependencies(name='bitsandbytes', upper_constraint='0.42'), # 0.41 works with CUDA 11.8