Skip to content

Commit

Permalink
Apply suggestions from code review
Browse files Browse the repository at this point in the history
Co-authored-by: Torsten Kilias <[email protected]>
  • Loading branch information
MarleneKress79789 and tkilias authored Feb 20, 2024
1 parent 2281e9a commit 9b36191
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ class BaseModelUDF(ABC):
This base class should be extended by each UDF class containing model logic.
This class contains common operations for all prediction UDFs:
- accesses data part-by-part based on predefined batch size
- manages the script cache
- manages the model cache
- reads the corresponding model from BucketFS into cache
- creates model pipeline through transformer api
- manages the creation of predictions and the preparation of results.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ class ModelDownloaderUDF:
"""
UDF which downloads a pretrained model from Huggingface using Huggingface's transformers API,
and uploads it to the BucketFS at , from where it can then be loaded without accessing Huggingface again.
Must be called with the following Input Table:
Must be called with the following Input Parameter:
model_name | sub_dir | bfs_conn | token_conn
---------------------------------------------------------------------------------------------------
Expand Down

0 comments on commit 9b36191

Please sign in to comment.