mirror/OpenLLM
1
0
Fork 0
mirror of https://github.com/bentoml/OpenLLM.git synced 2026-04-23 16:39:48 -04:00
Code Issues Packages Projects Releases Wiki Activity

infra: prepare for 0.0.8 release

Browse Source 
Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
This commit is contained in:
Aaron
2023-05-27 23:42:06 -07:00
parent 1e254a4cf3
commit 6aca092bfc
4 changed files with 10 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/openllm/__init__.py
View File

@@ -117,6 +117,7 @@ if t.TYPE_CHECKING:
from . import client as client
from . import exceptions as exceptions
from . import models as models
# Specific types import
from ._configuration import LLMConfig as LLMConfig
from ._llm import LLM as LLM
@@ -172,8 +173,7 @@ if t.TYPE_CHECKING:
from .utils.dummy_flax_objects import *
else:
from .models.auto import MODEL_FLAX_MAPPING as MODEL_FLAX_MAPPING
from .models.auto import \
MODEL_FLAX_MAPPING_NAMES as MODEL_FLAX_MAPPING_NAMES
from .models.auto import MODEL_FLAX_MAPPING_NAMES as MODEL_FLAX_MAPPING_NAMES
from .models.auto import AutoFlaxLLM as AutoFlaxLLM
from .models.flan_t5 import FlaxFlanT5 as FlaxFlanT5
@@ -184,8 +184,7 @@ if t.TYPE_CHECKING:
from .utils.dummy_tf_objects import *
else:
from .models.auto import MODEL_TF_MAPPING as MODEL_TF_MAPPING
from .models.auto import \
MODEL_TF_MAPPING_NAMES as MODEL_TF_MAPPING_NAMES
from .models.auto import MODEL_TF_MAPPING_NAMES as MODEL_TF_MAPPING_NAMES
from .models.auto import AutoTFLLM as AutoTFLLM
from .models.flan_t5 import TFFlanT5 as TFFlanT5

6
src/openllm/models/chatglm/__init__.py
View File

@@ -31,10 +31,8 @@ else:
_import_structure["modeling_chatglm"] = ["ChatGLM"]
if t.TYPE_CHECKING:
from .configuration_chatglm import \
DEFAULT_PROMPT_TEMPLATE as DEFAULT_PROMPT_TEMPLATE
from .configuration_chatglm import \
START_CHATGLM_COMMAND_DOCSTRING as START_CHATGLM_COMMAND_DOCSTRING
from .configuration_chatglm import DEFAULT_PROMPT_TEMPLATE as DEFAULT_PROMPT_TEMPLATE
from .configuration_chatglm import START_CHATGLM_COMMAND_DOCSTRING as START_CHATGLM_COMMAND_DOCSTRING
from .configuration_chatglm import ChatGLMConfig as ChatGLMConfig
try:

6
src/openllm/models/falcon/__init__.py
View File

@@ -31,10 +31,8 @@ else:
_import_structure["modeling_falcon"] = ["Falcon"]
if t.TYPE_CHECKING:
from .configuration_falcon import \
DEFAULT_PROMPT_TEMPLATE as DEFAULT_PROMPT_TEMPLATE
from .configuration_falcon import \
START_FALCON_COMMAND_DOCSTRING as START_FALCON_COMMAND_DOCSTRING
from .configuration_falcon import DEFAULT_PROMPT_TEMPLATE as DEFAULT_PROMPT_TEMPLATE
from .configuration_falcon import START_FALCON_COMMAND_DOCSTRING as START_FALCON_COMMAND_DOCSTRING
from .configuration_falcon import FalconConfig as FalconConfig
try:

7
src/openllm/utils/__init__.py
View File

@@ -29,15 +29,14 @@ import typing as t
import bentoml
import inflection
from bentoml._internal.types import LazyType as LazyType
# NOTE: The following exports useful utils from bentoml
from bentoml._internal.utils import LazyLoader as LazyLoader
from bentoml._internal.utils import bentoml_cattr as bentoml_cattr
from bentoml._internal.utils import \
copy_file_to_fs_folder as copy_file_to_fs_folder
from bentoml._internal.utils import copy_file_to_fs_folder as copy_file_to_fs_folder
from bentoml._internal.utils import pkg as pkg
from bentoml._internal.utils import reserve_free_port as reserve_free_port
from bentoml._internal.utils import \
resolve_user_filepath as resolve_user_filepath
from bentoml._internal.utils import resolve_user_filepath as resolve_user_filepath
from . import analytics as analytics
from . import codegen as codegen