






'''simple docstring'''
from typing import List, Optional, Tuple, Union

import torch

from ...models import UNetaDModel
from ...schedulers import PNDMScheduler
from ...utils import randn_tensor
from ..pipeline_utils import DiffusionPipeline, ImagePipelineOutput


class    _snake_case   (				snake_case					):

					"""simple docstring"""





					_UpperCamelCase		 =      42
					_UpperCamelCase		 =      42




					def __init__(     self   , UpperCAmelCase__   , UpperCAmelCase__     )      ->	int:
										super().__init__()

										a_									=   PNDMScheduler.from_config(scheduler.config     )

										self.register_modules(unet=UpperCAmelCase__   , scheduler=UpperCAmelCase__     )




					@torch.no_grad()
					def __call__(     self   , UpperCAmelCase__ = 1   , UpperCAmelCase__ = 50   , UpperCAmelCase__ = None   , UpperCAmelCase__ = "pil"   , UpperCAmelCase__ = True   , **UpperCAmelCase__   , )      ->	Union[ImagePipelineOutput, Tuple]:
										a_									=   randn_tensor(
										    (batch_size, self.unet.config.in_channels, self.unet.config.sample_size, self.unet.config.sample_size)   , generator=UpperCAmelCase__   , device=self.device   , )

										self.scheduler.set_timesteps(UpperCAmelCase__     )
										for t in self.progress_bar(self.scheduler.timesteps     ):
															a_									=   self.unet(UpperCAmelCase__   , UpperCAmelCase__     ).sample

															a_									=   self.scheduler.step(UpperCAmelCase__   , UpperCAmelCase__   , UpperCAmelCase__     ).prev_sample

										a_									=   (image / 2 + 0.5).clamp(0   , 1     )
										a_									=   image.cpu().permute(0   , 2   , 3   , 1     ).numpy()
										if output_type == "pil":
															a_									=   self.numpy_to_pil(UpperCAmelCase__     )

										if not return_dict:
															return (image,)

										return ImagePipelineOutput(images=UpperCAmelCase__     )









'''simple docstring'''
import os
import shutil
import unittest
from unittest.mock import patch

from transformers.testing_utils import CaptureStd, is_pt_tf_cross_test, require_torch


class    _snake_case   (				unittest.TestCase					):

					"""simple docstring"""





					@patch('sys.argv'   , ['fakeprogrampath', 'env']     )
					def  __SCREAMING_SNAKE_CASE (     self     )      ->	Tuple:
										# test transformers-cli env
										import transformers.commands.transformers_cli

										with CaptureStd() as cs:
															transformers.commands.transformers_cli.main()
										self.assertIn('Python version'   , cs.out     )
										self.assertIn('Platform'   , cs.out     )
										self.assertIn('Using distributed or parallel set-up in script?'   , cs.out     )




					@is_pt_tf_cross_test
					@patch(
					    'sys.argv'   , ['fakeprogrampath', 'pt-to-tf', '--model-name', 'hf-internal-testing/tiny-random-gptj', '--no-pr']     )
					def  __SCREAMING_SNAKE_CASE (     self     )      ->	List[str]:
										import transformers.commands.transformers_cli

										shutil.rmtree('/tmp/hf-internal-testing/tiny-random-gptj'   , ignore_errors=UpperCAmelCase__     )  # cleans potential past runs
										transformers.commands.transformers_cli.main()

										# The original repo has no TF weights -- if they exist, they were created by the CLI
										self.assertTrue(os.path.exists('/tmp/hf-internal-testing/tiny-random-gptj/tf_model.h5'     )     )




					@require_torch
					@patch('sys.argv'   , ['fakeprogrampath', 'download', 'hf-internal-testing/tiny-random-gptj', '--cache-dir', '/tmp']     )
					def  __SCREAMING_SNAKE_CASE (     self     )      ->	List[Any]:
										import transformers.commands.transformers_cli

										# # remove any previously downloaded model to start clean
										shutil.rmtree('/tmp/models--hf-internal-testing--tiny-random-gptj'   , ignore_errors=UpperCAmelCase__     )

										# run the command
										transformers.commands.transformers_cli.main()

										# check if the model files are downloaded correctly on /tmp/models--hf-internal-testing--tiny-random-gptj
										self.assertTrue(os.path.exists('/tmp/models--hf-internal-testing--tiny-random-gptj/blobs'     )     )
										self.assertTrue(os.path.exists('/tmp/models--hf-internal-testing--tiny-random-gptj/refs'     )     )
										self.assertTrue(os.path.exists('/tmp/models--hf-internal-testing--tiny-random-gptj/snapshots'     )     )




					@require_torch
					@patch(
					    'sys.argv'   , [
					        'fakeprogrampath',
					        'download',
					        'hf-internal-testing/test_dynamic_model_with_tokenizer',
					        '--trust-remote-code',
					        '--cache-dir',
					        '/tmp',
					    ]   , )
					def  __SCREAMING_SNAKE_CASE (     self     )      ->	int:
										import transformers.commands.transformers_cli

										# # remove any previously downloaded model to start clean
										shutil.rmtree('/tmp/models--hf-internal-testing--test_dynamic_model_with_tokenizer'   , ignore_errors=UpperCAmelCase__     )

										# run the command
										transformers.commands.transformers_cli.main()

										# check if the model files are downloaded correctly on /tmp/models--hf-internal-testing--test_dynamic_model_with_tokenizer
										self.assertTrue(os.path.exists('/tmp/models--hf-internal-testing--test_dynamic_model_with_tokenizer/blobs'     )     )
										self.assertTrue(os.path.exists('/tmp/models--hf-internal-testing--test_dynamic_model_with_tokenizer/refs'     )     )
										self.assertTrue(
										    os.path.exists('/tmp/models--hf-internal-testing--test_dynamic_model_with_tokenizer/snapshots'     )     )

