search

aws / sagemaker-huggingface-inference-toolkit

Apache License 2.0
235 stars 60 forks source link

Add support for additional tasks via HuggingFace TaskManager #113

Closed samruds closed 6 months ago

samruds commented 7 months ago

Issue #, if available: https://github.com/aws/sagemaker-huggingface-inference-toolkit/issues/112

Description of changes:

  1. Check for os variable set with task
  2. Update error message to indicate using inference.py if Transformers Tasks or Architecture_2_Task do not support this task type

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.