self.device in huggingface.py line 210 treated as torch.device but might be a string (#1172)
* self.device in huggingface.py line 210
In huggingface.py line 210, self.device is str and does not have a "type" attribute
* Update huggingface.py
This handles both the case where `self.device` is a `torch.device` and a string
* Update huggingface.py
---------
Co-authored-by:
Hailey Schoelkopf <65563625+haileyschoelkopf@users.noreply.github.com>
Showing
Please register or sign in to comment