|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from typing import TYPE_CHECKING |
|
|
|
from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available |
|
|
|
|
|
_import_structure = { |
|
"configuration_decision_transformer": [ |
|
"DECISION_TRANSFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP", |
|
"DecisionTransformerConfig", |
|
], |
|
} |
|
|
|
try: |
|
if not is_torch_available(): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
pass |
|
else: |
|
_import_structure["modeling_decision_transformer"] = [ |
|
"DECISION_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LIST", |
|
"DecisionTransformerGPT2Model", |
|
"DecisionTransformerGPT2PreTrainedModel", |
|
"DecisionTransformerModel", |
|
"DecisionTransformerPreTrainedModel", |
|
] |
|
|
|
|
|
if TYPE_CHECKING: |
|
from .configuration_decision_transformer import ( |
|
DECISION_TRANSFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP, |
|
DecisionTransformerConfig, |
|
) |
|
|
|
try: |
|
if not is_torch_available(): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
pass |
|
else: |
|
from .modeling_decision_transformer import ( |
|
DECISION_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LIST, |
|
DecisionTransformerGPT2Model, |
|
DecisionTransformerGPT2PreTrainedModel, |
|
DecisionTransformerModel, |
|
DecisionTransformerPreTrainedModel, |
|
) |
|
|
|
|
|
else: |
|
import sys |
|
|
|
sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) |
|
|