o x¡·i?ã@sFddlmZddlmZddlmZddlmZmZGdd„dƒZ dS) é)ÚSequence)ÚAny)ÚResponseé)ÚSUPPORTED_MULTIMODAL_PIPELINESÚ#DiffusersMultiModalPipelineResolverc @sHeZdZdZd dd„Zdeedeeefde d e d e def d d „Z dS)ÚDiffusersPipelineResolvera$Resolver for `DiffusionPipeline` request and responses from [HuggingFace Diffusers](https://huggingface.co/docs/diffusers/index), providing necessary data transformations, formatting, and logging. This is based off `wandb.sdk.integration_utils.auto_logging.RequestResponseResolver`. ÚreturnNcCsd|_d|_dS)Nr)Ú wandb_tableÚpipeline_call_count)Úself©r úa/home/ubuntu/.local/lib/python3.10/site-packages/wandb/integration/diffusers/pipeline_resolver.pyÚ__init__s z"DiffusersPipelineResolver.__init__ÚargsÚkwargsÚresponseÚ start_timeÚ time_elapsedc CsF|djj}d}|tvrt||jƒ}|jd7_||||||ƒ}|S)a1Main call method for the `DiffusersPipelineResolver` class. Args: args: (Sequence[Any]) List of arguments. kwargs: (Dict[str, Any]) Dictionary of keyword arguments. response: (wandb.sdk.integration_utils.auto_logging.Response) The response from the request. start_time: (float) Time when request started. time_elapsed: (float) Time elapsed for the request. Returns: Packed data as a dictionary for logging to wandb, None if an exception occurred. rNr)Ú __class__Ú__name__rrr ) r rrrrrÚ pipeline_nameÚresolverÚ loggable_dictr r rÚ__call__s ÿz"DiffusersPipelineResolver.__call__)r N) rÚ __module__Ú __qualname__Ú__doc__rrrÚdictÚstrrÚfloatrr r r rr s  þ ýüûúùrN) Úcollections.abcrÚtypingrÚ(wandb.sdk.integration_utils.auto_loggingrÚ resolversrrrr r r rÚs