tnadav
d3bf847e6f
* Fix type annotation for `load` parameter * Add type annotations to all `load` parameters * Avoid using mutable types for `load` default parameters * Add return type annotation to `load_tokenizer` * Export public module attributes
21 lines
377 B
Python
21 lines
377 B
Python
# Copyright © 2023-2024 Apple Inc.
|
|
|
|
import os
|
|
|
|
from ._version import __version__
|
|
|
|
os.environ["TRANSFORMERS_NO_ADVISORY_WARNINGS"] = "1"
|
|
|
|
from .convert import convert
|
|
from .generate import batch_generate, generate, stream_generate
|
|
from .utils import load
|
|
|
|
__all__ = [
|
|
"__version__",
|
|
"convert",
|
|
"batch_generate",
|
|
"generate",
|
|
"stream_generate",
|
|
"load",
|
|
]
|