|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from typing import TYPE_CHECKING |
|
|
|
|
|
from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available |
|
|
|
|
|
_import_structure = { |
|
"configuration_autoformer": [ |
|
"AUTOFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP", |
|
"AutoformerConfig", |
|
], |
|
} |
|
|
|
try: |
|
if not is_torch_available(): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
pass |
|
else: |
|
_import_structure["modeling_autoformer"] = [ |
|
"AUTOFORMER_PRETRAINED_MODEL_ARCHIVE_LIST", |
|
"AutoformerForPrediction", |
|
"AutoformerModel", |
|
"AutoformerPreTrainedModel", |
|
] |
|
|
|
|
|
if TYPE_CHECKING: |
|
from .configuration_autoformer import ( |
|
AUTOFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP, |
|
AutoformerConfig, |
|
) |
|
|
|
try: |
|
if not is_torch_available(): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
pass |
|
else: |
|
from .modeling_autoformer import ( |
|
AUTOFORMER_PRETRAINED_MODEL_ARCHIVE_LIST, |
|
AutoformerForPrediction, |
|
AutoformerModel, |
|
AutoformerPreTrainedModel, |
|
) |
|
|
|
else: |
|
import sys |
|
|
|
sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) |
|
|