


"""simple docstring"""


# Copyright 2023 The HuggingFace Inc. team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from ..models.whisper import WhisperForConditionalGeneration, WhisperProcessor
from .base import PipelineTool
class 						_UpperCamelCase						(     lowerCAmelCase__	):
							'''simple docstring'''







							__UpperCAmelCase		:					Any          ="""openai/whisper-base"""
							__UpperCAmelCase		:					List[str]          =(
							    """This is a tool that transcribes an audio into text. It takes an input named `audio` and returns the """
							    """transcribed text."""
							)
							__UpperCAmelCase		:					Optional[Any]          ="""transcriber"""
							__UpperCAmelCase		:					Any          =WhisperProcessor
							__UpperCAmelCase		:					Dict          =WhisperForConditionalGeneration

							__UpperCAmelCase		:					Dict          =["""audio"""]
							__UpperCAmelCase		:					Optional[int]          =["""text"""]

							def     snake_case       (  self					,     __a			):
										return self.pre_processor(__a					,     return_tensors="pt"			).input_features

							def     snake_case       (  self					,     __a			):
										return self.model.generate(inputs=__a			)



							def     snake_case       (  self					,     __a			):
										return self.pre_processor.batch_decode(__a					,     skip_special_tokens=__a			)[0]








"""simple docstring"""


import gc
import threading
import time

import psutil
import torch
class 						_UpperCamelCase						:
							'''simple docstring'''







							def __init__(  self			):
										__lowerCAmelCase       					=    psutil.Process()
										__lowerCAmelCase       					=    False

							def     snake_case       (  self			):
										__lowerCAmelCase       					=    -1

										while True:
													__lowerCAmelCase       					=    max(self.process.memory_info().rss					,     self.cpu_memory_peak			)

													# can't sleep or will not catch the peak right (this comment is here on purpose)
													if not self.peak_monitoring:
																break

							def     snake_case       (  self			):
										__lowerCAmelCase       					=    True
										__lowerCAmelCase       					=    threading.Thread(target=self.peak_monitor			)
										__lowerCAmelCase       					=    True
										self.thread.start()



							def     snake_case       (  self			):
										__lowerCAmelCase       					=    False
										self.thread.join()
										return self.cpu_memory_peak
A  :     Any    					=    PeakCPUMemory()


def _lowerCamelCase			(	):

			'''simple docstring'''




			__lowerCAmelCase       					=    {"time": time.time()}

			gc.collect()
			torch.cuda.empty_cache()

			# CPU mem
			__lowerCAmelCase       					=    psutil.Process().memory_info().rss
			cpu_peak_tracker.start()

			# GPU mem
			for i in range(torch.cuda.device_count()   ):
						__lowerCAmelCase       					=    torch.cuda.memory_allocated(_UpperCamelCase   )
			torch.cuda.reset_peak_memory_stats()

			return measures


def _lowerCamelCase			(	_UpperCamelCase   ):

			'''simple docstring'''




			__lowerCAmelCase       					=    {"time": time.time() - start_measures["time"]}

			gc.collect()
			torch.cuda.empty_cache()

			# CPU mem
			__lowerCAmelCase       					=    (psutil.Process().memory_info().rss - start_measures["cpu"]) / 2**20
			__lowerCAmelCase       					=    (cpu_peak_tracker.stop() - start_measures["cpu"]) / 2**20

			# GPU mem
			for i in range(torch.cuda.device_count()   ):
						__lowerCAmelCase       					=    (torch.cuda.memory_allocated(_UpperCamelCase   ) - start_measures[str(_UpperCamelCase   )]) / 2**20
						__lowerCAmelCase       					=    (torch.cuda.max_memory_allocated(_UpperCamelCase   ) - start_measures[str(_UpperCamelCase   )]) / 2**20

			return measures


def _lowerCamelCase			(	_UpperCamelCase					,				_UpperCamelCase   ):

			'''simple docstring'''




			print(f"{description}:"   )
			print(f"- Time: {measures['time']:.2f}s"   )
			for i in range(torch.cuda.device_count()   ):
						print(f"- GPU {i} allocated: {measures[str(_UpperCamelCase   )]:.2f}MiB"   )
						__lowerCAmelCase       					=    measures[f"{i}-peak"]
						print(f"- GPU {i} peak: {peak:.2f}MiB"   )
			print(f"- CPU RAM allocated: {measures['cpu']:.2f}MiB"   )
			print(f"- CPU RAM peak: {measures['cpu-peak']:.2f}MiB"   )




