huggingface.py 443 B

123456789101112131415161718
  1. import logging
  2. from wandb.sdk.integration_utils.auto_logging import AutologAPI
  3. from .resolver import HuggingFacePipelineRequestResponseResolver
  4. logger = logging.getLogger(__name__)
  5. resolver = HuggingFacePipelineRequestResponseResolver()
  6. autolog = AutologAPI(
  7. name="transformers",
  8. symbols=("Pipeline.__call__",),
  9. resolver=resolver,
  10. telemetry_feature="hf_pipeline_autolog",
  11. )
  12. autolog.get_latest_id = resolver.get_latest_id