jamtur01's picture
Upload folder using huggingface_hub
9c6594c verified
raw
history blame contribute delete
443 Bytes
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