Lightning-Universe / lightning-transformers

Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning

https://lightning-transformers.readthedocs.io

Apache License 2.0

610 stars 77 forks source link

Add ability to set kwargs for pipeline in load_from_checkpoint #204

Closed SeanNaren closed 2 years ago

SeanNaren commented 2 years ago

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[bot] commented 2 years ago

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