Skip to content

Replace 'cuda()' with 'to(device)'.#18

Open
minigb wants to merge 1 commit intonaver-ai:mainfrom
minigb:Fix/Support-non-cuda
Open

Replace 'cuda()' with 'to(device)'.#18
minigb wants to merge 1 commit intonaver-ai:mainfrom
minigb:Fix/Support-non-cuda

Conversation

@minigb
Copy link

@minigb minigb commented Nov 14, 2023

Hi,

I found a part that was using cuda() and have updated it to to(device) to ensure compatibility with non-CUDA devices.
Please take a look.

Thank you!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant

Comments