a þd–ã@s<ddlmZmZddlmZmZddlZGdd„deƒZdS)é)ÚABCÚabstractmethod)ÚAnyÚDictNc@s¨eZdZdZeejddœdd„ƒZeddœdd„ƒZe ee e d œd d „ƒƒZ e ee e d œd d „ƒƒZ e ee dœdd„ƒƒZe eedœdd„ƒƒZeeddœdd„ƒZdS)Ú Acceleratorz`The Accelerator base class. An Accelerator is meant to deal with one type of hardware. N)ÚdeviceÚreturncCsdS)z6Create and prepare the device for the current process.N©)Úselfrr r úr/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/lightning_fabric/accelerators/accelerator.pyÚ setup_deviceszAccelerator.setup_device)rcCsdS)z.Clean up any state created by the accelerator.Nr )r r r r ÚteardownszAccelerator.teardown)ÚdevicesrcCsdS)z!Accelerator device parsing logic.Nr ©rr r r Ú parse_devices"szAccelerator.parse_devicescCsdS)z*Gets parallel devices for the Accelerator.Nr rr r r Úget_parallel_devices'sz Accelerator.get_parallel_devicescCsdS)z&Get the device count when set to auto.Nr r r r r Úauto_device_count,szAccelerator.auto_device_countcCsdS)z$Detect if the hardware is available.Nr r r r r Ú is_available1szAccelerator.is_available)Úaccelerator_registryrcCsdS)Nr )Úclsrr r r Úregister_accelerators6sz!Accelerator.register_accelerators)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚtorchrr r Ú staticmethodrrrÚintrÚboolrÚ classmethodrrr r r r rs&r)ÚabcrrÚtypingrrrrr r r r Ús