Skip to content

allow string device input for OrtModel.to(device)#337

Merged
regisss merged 4 commits intohuggingface:mainfrom
jessecambon:string-device-input
Aug 8, 2022
Merged

allow string device input for OrtModel.to(device)#337
regisss merged 4 commits intohuggingface:mainfrom
jessecambon:string-device-input

Conversation

@jessecambon
Copy link
Copy Markdown
Contributor

What does this PR do?

Implements #336

Copy link
Copy Markdown
Contributor

@regisss regisss left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for adding this @jessecambon!!

@regisss
Copy link
Copy Markdown
Contributor

regisss commented Aug 8, 2022

@jessecambon You need to rebase your branch on the main branch so that the tests pass

@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

HuggingFaceDocBuilderDev commented Aug 8, 2022

The documentation is not available anymore as the PR was closed or merged.

@jessecambon jessecambon force-pushed the string-device-input branch from 133377b to 8c33e0f Compare August 8, 2022 14:59
@jessecambon
Copy link
Copy Markdown
Contributor Author

@regisss thanks, I also added the same modification to the sequence to sequence model and added tests. My fork should now be up-to-date with main.

@regisss
Copy link
Copy Markdown
Contributor

regisss commented Aug 8, 2022

@regisss thanks, I also added the same modification to the sequence to sequence model and added tests. My fork should now be up-to-date with main.

Thanks a lot @jessecambon 😃
It all looks good to me. I just launched the tests and if they all pass I will merge this PR.

@regisss regisss merged commit f69264f into huggingface:main Aug 8, 2022
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.

3 participants