Huggingface model wrapper update

[GiulioZizzo]

Description

Currently, the HF estimator expects a HF model that is on the cpu: if a user moves the model to the GPU before giving it to the estimator there will be a crash to to a mismatch between the tensors created for the forward hooks in the HF model wrapper and the model. This small update ensures that the correct device is used between model and hook inputs.

Fixes # TO CREATE

Type of change

Please check all relevant options.

• Improvement (non-breaking)
• Bug fix (non-breaking)
• New feature (non-breaking)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• This change requires a documentation update

Testing

As the CI tests are CPU limited we have tested this via updating the notebook to shift the models to GPU devices in a few places before feeding to the estimator.

Test Configuration:

• OS: Linux and Mac
• Python version: 3.9
• ART version or commit number: 1.16
• TensorFlow / Keras / PyTorch / MXNet version: torch 2.0

Checklist

• My code follows the style guidelines of this project
• I have performed a self-review of my own code
• I have commented my code
• I have made corresponding changes to the documentation
• My changes generate no new warnings
• I have added tests that prove my fix is effective or that my feature works
• New and existing unit tests pass locally with my changes
• My changes have been tested using both CPU and GPU devices

[codecov-commenter]

Codecov Report

Merging #2300 (e896821) into dev_1.17.0 (ab389e7) will decrease coverage by 1.76%.
Report is 4 commits behind head on dev_1.17.0.
The diff coverage is 0.00%.

❗ Your organization needs to install the Codecov GitHub app to enable full functionality.

Additional details and impacted files

@@              Coverage Diff               @@
##           dev_1.17.0    #2300      +/-   ##
==============================================
- Coverage       85.60%   83.84%   -1.76%     
==============================================
  Files             324      324              
  Lines           29326    29982     +656     
  Branches         5407     5538     +131     
==============================================
+ Hits            25104    25138      +34     
- Misses           2840     3448     +608     
- Partials         1382     1396      +14     

Files	Coverage Δ
art/estimators/classification/hugging_face.py	90.59% <0.00%> (-2.39%)	⬇️

... and 7 files with indirect coverage changes

[beat-buesser]

Hi @GiulioZizzo Thank you very much for your pull request! Looks good to me.