File size: 415 Bytes
e05eed1
98a67a0
 
 
 
 
 
 
 
 
 
ea7747b
1
2
3
4
5
6
7
8
9
10
11
12
# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

import importlib

# Ensure torch is imported so its shared libs (e.g., libc10.so) are available
try:
    import torch  # noqa: F401
except Exception:  # torch may be cpu-only/cuda-only, still attempt import of extension
    pass

from ._nemotron_ocr_cpp import *  # noqa: F403