Skip to content
New issue

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

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

Already on GitHub? # to your account

Fix cpu #22

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Fix cpu #22

wants to merge 1 commit into from

Conversation

edumotya
Copy link

Fix #21 by simply eliminating the explicit conversion of zero_modules to cuda.

As far as I see create_model will take care of moving everything to cuda

model.to(device=device)

@LukaGiorgadze
Copy link

Thank you, tiny but helpful PR.
Did you test on cpu?

@edumotya
Copy link
Author

Sure! it works

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

how to use cpu/mps?
3 participants