File size: 1,625 Bytes
4d9b003 be62f78 4d9b003 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | # SPDX-License-Identifier: Apache-2.0
"""Diffusion Planner v5.0 (Autoware diffusion_planner) on a Tenstorrent Blackhole p150 (tt-nn), packaged as
changh95/diffusion-planner-p150.
Python API (see PYTHON.md)::
from tt_diffusion_planner import DiffusionPlanner
with DiffusionPlanner.from_pretrained(device_id=0) as model:
out = model(inputs="code/tt_diffusion_planner/samples/kashiwanoha_dense.npz")
Importing the package has no side effects (no device, no ttnn / torch import, no network); the names below load
on first use. The HTTP server is ``tt_diffusion_planner.server.app:app`` (SERVING.md). ``tt_diffusion_planner.ttaw``
is the vendored shared package of the Autoware ports (``ttaw/VENDORED.json`` records its version and file hashes;
never edit it here). ``tt_diffusion_planner.reference`` is the fp32 CPU reference (torch) and
``tt_diffusion_planner.host`` the node's pre- and post-processing (numpy).
"""
__version__ = "0.2.0"
__all__ = ["DiffusionPlanner", "Output", "open_device", "load_inputs", "INPUT_SCHEMA", "__version__"]
_LAZY = {
"DiffusionPlanner": (".api", "DiffusionPlanner"),
"Output": (".api", "Output"),
"open_device": (".device", "open_device"),
"load_inputs": (".io", "load_inputs"),
"INPUT_SCHEMA": (".reference.config", "INPUT_SCHEMA"),
}
def __getattr__(name):
if name in _LAZY:
import importlib
module, attr = _LAZY[name]
return getattr(importlib.import_module(module, __name__), attr)
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
def __dir__():
return sorted(list(globals()) + __all__)
|