Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

GPT Generation error on GPU #57

Closed
jalammar opened this issue Dec 31, 2021 · 1 comment
Closed

GPT Generation error on GPU #57

jalammar opened this issue Dec 31, 2021 · 1 comment
Labels
bug Something isn't working

Comments

@jalammar
Copy link
Owner

Generating text with GPT2 on GPU throws the following error:

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument index in method wrapper__index_select)
@jalammar jalammar added the bug Something isn't working label Dec 31, 2021
@JoaoLages
Copy link
Contributor

JoaoLages commented Jan 3, 2022 •
edited
Loading

Solving this issue in here: https://github.com/JoaoLages/ecco/tree/bugfix/issue_%2357

@JoaoLages JoaoLages mentioned this issue Jan 3, 2022
Merged
jalammar added a commit that referenced this issue Jan 3, 2022
@jalammar
Merge pull request #59 from JoaoLages/bugfix/issue_#57
9aa8639 
Bugfix/issue #57
@jalammar jalammar closed this as completed Jan 4, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
bug Something isn't working
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@jalammar @JoaoLages