File size: 443 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import logging
from wandb.sdk.integration_utils.auto_logging import AutologAPI
from .resolver import HuggingFacePipelineRequestResponseResolver
logger = logging.getLogger(__name__)
resolver = HuggingFacePipelineRequestResponseResolver()
autolog = AutologAPI(
name="transformers",
symbols=("Pipeline.__call__",),
resolver=resolver,
telemetry_feature="hf_pipeline_autolog",
)
autolog.get_latest_id = resolver.get_latest_id
|