huggingface / huggingface/accelerate
gather objects in TPU is not supported
Open
feature request
TPU
- Dominant language
- Python
- Stars
- 9.9k
- Forks
- 1.5k
- Avg merge
- 5d 2h
- Merged PRs (30d)
- 27
Description
Do you have any plans to implement the above functionality on a TPU (Also is this even possible on TPU ?) ? I'm trying to distribute evaluation using accelerate.
Error:
```sh
raise NotImplementedError("gather objects in TPU is not supported")
NotImplementedError: gather objects in TPU is not supported
```
Contributor guide
Assessment
This issue has not been assessed yet.