a d @s8ddlmZddlmZddlmZGdddeZdS))Any) Precision) Optimizablec@s"eZdZdZeeedddZdS) TPUPrecisionz%Precision plugin for TPU integration.) optimizerkwargsreturncKs ddlmm}|j||dS)Nr)Zoptimizer_args)Ztorch_xla.core.xla_modelcoreZ xla_modeloptimizer_step)selfrrZxmr o/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/lightning_fabric/plugins/precision/tpu.pyr szTPUPrecision.optimizer_stepN)__name__ __module__ __qualname____doc__rrr r r r r rs rN)typingrZ,lightning_fabric.plugins.precision.precisionrZ lightning_fabric.utilities.typesrrr r r r s