optimum
07cb52d6 - Load model directly on cuda device for the ONNX export (#1112)

Commit

2 years ago

Load model directly on cuda device for the ONNX export (#1112) * load directly on cuda device if necessary * fix import

References

Author

fxmarty

Parents