Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Move N-term tokens to the correct device
The device changes between initialization and actual running, hence doing this in __init__ is insufficient.
- Loading branch information