|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from typing import TYPE_CHECKING |
|
|
|
from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available |
|
|
|
|
|
_import_structure = { |
|
"configuration_bigbird_pegasus": [ |
|
"BIGBIRD_PEGASUS_PRETRAINED_CONFIG_ARCHIVE_MAP", |
|
"BigBirdPegasusConfig", |
|
"BigBirdPegasusOnnxConfig", |
|
], |
|
} |
|
|
|
try: |
|
if not is_torch_available(): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
pass |
|
else: |
|
_import_structure["modeling_bigbird_pegasus"] = [ |
|
"BIGBIRD_PEGASUS_PRETRAINED_MODEL_ARCHIVE_LIST", |
|
"BigBirdPegasusForCausalLM", |
|
"BigBirdPegasusForConditionalGeneration", |
|
"BigBirdPegasusForQuestionAnswering", |
|
"BigBirdPegasusForSequenceClassification", |
|
"BigBirdPegasusModel", |
|
"BigBirdPegasusPreTrainedModel", |
|
] |
|
|
|
|
|
if TYPE_CHECKING: |
|
from .configuration_bigbird_pegasus import ( |
|
BIGBIRD_PEGASUS_PRETRAINED_CONFIG_ARCHIVE_MAP, |
|
BigBirdPegasusConfig, |
|
BigBirdPegasusOnnxConfig, |
|
) |
|
|
|
try: |
|
if not is_torch_available(): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
pass |
|
else: |
|
from .modeling_bigbird_pegasus import ( |
|
BIGBIRD_PEGASUS_PRETRAINED_MODEL_ARCHIVE_LIST, |
|
BigBirdPegasusForCausalLM, |
|
BigBirdPegasusForConditionalGeneration, |
|
BigBirdPegasusForQuestionAnswering, |
|
BigBirdPegasusForSequenceClassification, |
|
BigBirdPegasusModel, |
|
BigBirdPegasusPreTrainedModel, |
|
) |
|
|
|
|
|
else: |
|
import sys |
|
|
|
sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) |
|
|