Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Freddy Snijder's avatar
    Fix for Key Error that can happen in certain FSDP wrapping scenarios of... · e6acdcc3
    Freddy Snijder authored 
    Fix for Key Error that can happen in certain FSDP wrapping scenarios of Huggingface model sub-modules (issue #876) (#881)

* Fix for Key Error that can happen in certain FSDP wrapping scenarios of Huggingface model sub-modules (issue #876)

* Styling fixes

* Updated the test to be independent of the Huggingface transformers package

* Added test for issue #876

* Small error message fix

* Skip test when CUDA is not available

* Fixed naming of model
    e6acdcc3
ci_test_list_3.txt 1018 Bytes