





"""simple docstring"""






import os
import sys

import transformers


lowerCAmelCase_			=							"3"

print('''Python version:''', sys.version)
print('''transformers version:''', transformers.__version__)

try:
    import torch

    print('''Torch version:''', torch.__version__)
    print('''Cuda available:''', torch.cuda.is_available())
    print('''Cuda version:''', torch.version.cuda)
    print('''CuDNN version:''', torch.backends.cudnn.version())
    print('''Number of GPUs available:''', torch.cuda.device_count())
    print('''NCCL version:''', torch.cuda.nccl.version())
except ImportError:
    print('''Torch version:''', None)

try:
    import deepspeed

    print('''DeepSpeed version:''', deepspeed.__version__)
except ImportError:
    print('''DeepSpeed version:''', None)

try:
    import tensorflow as tf

    print('''TensorFlow version:''', tf.__version__)
    print('''TF GPUs available:''', bool(tf.config.list_physical_devices('''GPU''')))
    print('''Number of TF GPUs available:''', len(tf.config.list_physical_devices('''GPU''')))
except ImportError:
    print('''TensorFlow version:''', None)













"""simple docstring"""






from dataclasses import dataclass
from typing import List, Optional, Union

import numpy as np
import PIL

from ...utils import BaseOutput, OptionalDependencyNotAvailable, is_torch_available, is_transformers_available
from .timesteps import (
    fastaa_timesteps,
    smartaa_timesteps,
    smartaa_timesteps,
    smartaaa_timesteps,
    smartaaa_timesteps,
    superaa_timesteps,
    superaa_timesteps,
    superaaa_timesteps,
)



@dataclass
class 	lowerCAmelCase					(      snake_case							):
       lowerCAmelCase__						 =  42
       lowerCAmelCase__						 =  42
       lowerCAmelCase__						 =  42


try:
    if not (is_transformers_available() and is_torch_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    from ...utils.dummy_torch_and_transformers_objects import *  # noqa F403
else:
    from .pipeline_if import IFPipeline
    from .pipeline_if_imgaimg import IFImgaImgPipeline
    from .pipeline_if_imgaimg_superresolution import IFImgaImgSuperResolutionPipeline
    from .pipeline_if_inpainting import IFInpaintingPipeline
    from .pipeline_if_inpainting_superresolution import IFInpaintingSuperResolutionPipeline
    from .pipeline_if_superresolution import IFSuperResolutionPipeline
    from .safety_checker import IFSafetyChecker
    from .watermark import IFWatermarker






