



'''simple docstring'''




import os
from typing import Dict, List, Tuple, TypeVar, Union


lowerCamelCase							:Union[str, Any]				    =					TypeVar('''T''')

lowerCamelCase							:List[Any]				    =					Union[List[T], Tuple[T, ...]]
lowerCamelCase							:Union[str, Any]				    =					Union[T, List[T], Dict[str, T]]
lowerCamelCase							:Optional[int]				    =					Union[str, bytes, os.PathLike]




'''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''')