File size: 573 Bytes
ba7051a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | # 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)
|