Skip to content
This repository has been archived by the owner on Nov 21, 2022. It is now read-only.

Add ability to set kwargs for pipeline in load_from_checkpoint #204

Merged
merged 2 commits into from
Oct 20, 2021

Conversation

SeanNaren
Copy link
Contributor

Fixes a user issue where they'd like to set the params when loading the model. Added a rudimentary test, could clean it up via mocking but for now will suffice.

@codecov
Copy link

codecov bot commented Oct 20, 2021

Codecov Report

Merging #204 (328e225) into master (0852dde) will increase coverage by 0%.
The diff coverage is 100%.

@@          Coverage Diff          @@
##           master   #204   +/-   ##
=====================================
  Coverage      86%    86%           
=====================================
  Files          70     70           
  Lines        1542   1549    +7     
=====================================
+ Hits         1332   1339    +7     
  Misses        210    210           

@SeanNaren SeanNaren merged commit fc47034 into master Oct 20, 2021
@SeanNaren SeanNaren deleted the feat/hf_pipeline_kwargs branch October 20, 2021 23:03
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant