fix import

This commit is contained in:
hlky 2023-05-15 19:24:00 +01:00
parent eccba18c17
commit abba81d8a6

View File

@ -23,9 +23,9 @@ from typing import Callable, Dict, List, NamedTuple, Optional, Tuple, TypeVar, U
import numpy as np
from dtype import dtype_str_to_enum
from misc import is_linux, is_windows
from torch_utils import torch_dtype_to_string
from .dtype import dtype_str_to_enum
from .misc import is_linux, is_windows
from .torch_utils import torch_dtype_to_string
# Controls how many runtimes will be used in ModelContainer by default.
# See the runtime README.md for more information on the Model/ModelContainer