moge-2-p150 / code /tt_moge /reference /__init__.py
changh95's picture
Add Tenstorrent Blackhole tt-nn port
ba7051a verified
Raw History Blame Contribute Delete
573 Bytes
# SPDX-FileCopyrightText: (c) 2026 Tenstorrent AI ULC
# SPDX-License-Identifier: Apache-2.0
"""Torch reference for MoGe-2.
The faithful reference is the upstream `microsoft/MoGe` model (vendored here as
the ``moge`` package, with its ``utils3d`` dependency). Unlike a from-scratch
re-implementation, this guarantees the TT-NN port is graded against the exact
published model. ``load_pretrained.load_moge2`` returns a ready ``MoGeModel``.
"""
import os
import sys
_HERE = os.path.dirname(os.path.abspath(__file__))
if _HERE not in sys.path:
sys.path.insert(0, _HERE)