exasol / transformers-extension

An Exasol extension for using state-of-the-art pretrained machine learning models via the Hugging Face Transformers API.
MIT License
2 stars 2 forks source link

Add HuggingFaceHubBucketFSModelTransfer with save_pretrained #143

Closed MarleneKress79789 closed 10 months ago

MarleneKress79789 commented 1 year ago

Background:

Acceptance Criteria

### Tasks
- [x] Add a duplicate of the existing [model download](https://github.com/exasol/transformers-extension/blob/1321af15f1cf0e9ac68ca12c02a3d6e4318a2cb1/exasol_transformers_extension/udfs/models/model_downloader_udf.py) which uses the save_pretrained function and
- [x] add tests for the new download functionality
MarleneKress79789 commented 1 year ago

planning and splitting ticket: 0.5 pd