





from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_tokenizers_available, is_torch_available


_lowerCamelCase 					=      {
    'configuration_roc_bert': ['ROC_BERT_PRETRAINED_CONFIG_ARCHIVE_MAP', 'RoCBertConfig'],
    'tokenization_roc_bert': ['RoCBertTokenizer'],
}

try:
	if not is_tokenizers_available():
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	pass
else:
	pass

try:
	if not is_torch_available():
		raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
	pass
else:
	_lowerCamelCase 					=      [
	    'ROC_BERT_PRETRAINED_MODEL_ARCHIVE_LIST',
	    'RoCBertForCausalLM',
	    'RoCBertForMaskedLM',
	    'RoCBertForMultipleChoice',
	    'RoCBertForPreTraining',
	    'RoCBertForQuestionAnswering',
	    'RoCBertForSequenceClassification',
	    'RoCBertForTokenClassification',
	    'RoCBertLayer',
	    'RoCBertModel',
	    'RoCBertPreTrainedModel',
	    'load_tf_weights_in_roc_bert',
	]

if TYPE_CHECKING:
	from .configuration_roc_bert import ROC_BERT_PRETRAINED_CONFIG_ARCHIVE_MAP, RoCBertConfig
	from .tokenization_roc_bert import RoCBertTokenizer

	try:
		if not is_tokenizers_available():
			raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
		pass
	else:
		raise OptionalDependencyNotAvailable()

	try:
		if not is_torch_available():
			raise OptionalDependencyNotAvailable()
	except OptionalDependencyNotAvailable:
		pass
	else:
		from .modeling_roc_bert import (
		    ROC_BERT_PRETRAINED_MODEL_ARCHIVE_LIST,
		    RoCBertForCausalLM,
		    RoCBertForMaskedLM,
		    RoCBertForMultipleChoice,
		    RoCBertForPreTraining,
		    RoCBertForQuestionAnswering,
		    RoCBertForSequenceClassification,
		    RoCBertForTokenClassification,
		    RoCBertLayer,
		    RoCBertModel,
		    RoCBertPreTrainedModel,
		    load_tf_weights_in_roc_bert,
		)


else:
	import sys

	_lowerCamelCase 					=      _LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)


a_      =     frozenset(
    [
        'prompt',
        'height',
        'width',
        'guidance_scale',
        'negative_prompt',
        'prompt_embeds',
        'negative_prompt_embeds',
        'cross_attention_kwargs',
    ]
)

a_      =     frozenset(['prompt', 'negative_prompt'])

a_      =     frozenset([])

a_      =     frozenset(['image'])

a_      =     frozenset(
    [
        'image',
        'height',
        'width',
        'guidance_scale',
    ]
)

a_      =     frozenset(['image'])

a_      =     frozenset(
    [
        'prompt',
        'image',
        'height',
        'width',
        'guidance_scale',
        'negative_prompt',
        'prompt_embeds',
        'negative_prompt_embeds',
    ]
)

a_      =     frozenset(['prompt', 'image', 'negative_prompt'])

a_      =     frozenset(
    [
        # Text guided image variation with an image mask
        'prompt',
        'image',
        'mask_image',
        'height',
        'width',
        'guidance_scale',
        'negative_prompt',
        'prompt_embeds',
        'negative_prompt_embeds',
    ]
)

a_      =     frozenset(['prompt', 'image', 'mask_image', 'negative_prompt'])

a_      =     frozenset(
    [
        # image variation with an image mask
        'image',
        'mask_image',
        'height',
        'width',
        'guidance_scale',
    ]
)

a_      =     frozenset(['image', 'mask_image'])

a_      =     frozenset(
    [
        'example_image',
        'image',
        'mask_image',
        'height',
        'width',
        'guidance_scale',
    ]
)

a_      =     frozenset(['example_image', 'image', 'mask_image'])

a_      =     frozenset(['class_labels'])

a_      =     frozenset(['class_labels'])

a_      =     frozenset(['batch_size'])

a_      =     frozenset([])

a_      =     frozenset(['batch_size'])

a_      =     frozenset([])

a_      =     frozenset(
    [
        'prompt',
        'audio_length_in_s',
        'guidance_scale',
        'negative_prompt',
        'prompt_embeds',
        'negative_prompt_embeds',
        'cross_attention_kwargs',
    ]
)

a_      =     frozenset(['prompt', 'negative_prompt'])
a_      =     frozenset(['input_tokens'])

a_      =     frozenset(['input_tokens'])