optimum
Load model directly on cuda device for the ONNX export
#1112
Merged
Overview
Commits
2
Changes
View On GitHub

Load model directly on cuda device for the ONNX export #1112

fxmarty merged 2 commits into huggingface:main from fxmarty:lower-mem-usage-export
fxmarty
fxmarty load directly on cuda device if necessary
46f8969f
fxmarty fix import
7926fd51
HuggingFaceDocBuilderDev
fxmarty fxmarty merged 07cb52d6 into main 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum