



'''simple docstring'''




from typing import TYPE_CHECKING

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


lowerCamelCase							:Any				    =					{
    '''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							:List[Any]				    =					[
						    '''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							:Tuple				    =					_LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)




'''simple docstring'''




import torch

from diffusers import StableDiffusionPipeline


lowerCamelCase							:Tuple				    =					'''path-to-your-trained-model'''
lowerCamelCase							:Optional[int]				    =					StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.floataa).to('''cuda''')

lowerCamelCase							:Optional[int]				    =					'''A photo of sks dog in a bucket'''
lowerCamelCase							:List[Any]				    =					pipe(prompt, num_inference_steps=5_0, guidance_scale=7.5).images[0]

image.save('''dog-bucket.png''')