chore(qol): make envvar private

Signed-off-by: paperspace <29749331+aarnphm@users.noreply.github.com>
This commit is contained in:
paperspace
2024-05-27 18:07:24 +00:00
parent a4a6060f69
commit 9da0b4134c
2 changed files with 5 additions and 16 deletions

View File

@@ -1,5 +1,9 @@
{% extends bento_base_template %}
{% block SETUP_BENTO_COMPONENTS %}
ARG NVIDIA_DRIVER_CAPABILITIES=compute,utility
ENV NVIDIA_DRIVER_CAPABILITIES=$NVIDIA_DRIVER_CAPABILITIES
ARG HF_HUB_DISABLE_PROGRESS_BARS=TRUE
ENV HF_HUB_DISABLE_PROGRESS_BARS=$HF_HUB_DISABLE_PROGRESS_BARS
{% call common.RUN(__enable_buildkit__) -%} {{ __pip_cache__ }} {% endcall -%} bash -c 'pip install --no-color --progress-bar off "vllm==0.4.2" || true'
{{ super() }}
{% endblock %}

View File

@@ -17,13 +17,7 @@ from openllm_core.utils import (
get_quiet_mode,
normalise_model_name,
)
from openllm_core._typing_compat import (
LiteralQuantise,
LiteralSerialisation,
LiteralDtype,
get_literal_args,
TypedDict,
)
from openllm_core._typing_compat import LiteralQuantise, LiteralSerialisation, LiteralDtype, get_literal_args
from . import _termui as termui
if sys.version_info >= (3, 11):
@@ -356,11 +350,6 @@ def build_sdist(target_path: str, package: t.Literal['openllm', 'openllm-core',
return sdist_filename
class EnvironmentEntry(TypedDict):
name: str
value: str
@cli.command(name='build', context_settings={'token_normalize_func': inflection.underscore})
@shared_decorator
@click.option(
@@ -525,10 +514,6 @@ def build_command(
models=[ModelSpec.from_item({'tag': str(bentomodel.tag), 'alias': bentomodel.tag.name})]
if bentomodel is not None
else [],
envs=[
EnvironmentEntry(name='NVIDIA_DRIVER_CAPABILITIES', value='compute,utility'),
EnvironmentEntry(name=HF_HUB_DISABLE_PROGRESS_BARS, value='TRUE'),
],
description=service_readme,
include=list(llm_fs.walk.files()),
exclude=['/venv', '/.venv', '__pycache__/', '*.py[cod]', '*$py.class'],